Final_Assignment_Template

Runtime error

LamiaYT commited on Jun 28

Commit

25405da

1 Parent(s): f0b3f91

Fix Ver2

Files changed (1) hide show

app.py CHANGED Viewed

@@ -42,7 +42,8 @@ model = AutoModelForCausalLM.from_pretrained(
     trust_remote_code=True,
     torch_dtype=torch.float32  # Use float32 for CPU compatibility
 )
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 load_time = time.time() - start_time
 print(f"Model loaded in {load_time:.2f} seconds")

     trust_remote_code=True,
     torch_dtype=torch.float32  # Use float32 for CPU compatibility
 )
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_fast=False)
 load_time = time.time() - start_time
 print(f"Model loaded in {load_time:.2f} seconds")