Spaces:

krisha06
/

Python_tutor

Sleeping

krisha06 commited on Jun 23

Commit

2149ab2

verified ·

1 Parent(s): 9573ebf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ def load_model():
     tokenizer.pad_token = tokenizer.eos_token if tokenizer.pad_token is None else tokenizer.pad_token
     base_model = AutoModelForCausalLM.from_pretrained(
-        "TinyLLaMA/TinyLLaMA-1.1B-Chat-v1.0",
         torch_dtype=torch.float32,
     ).to(device)

     tokenizer.pad_token = tokenizer.eos_token if tokenizer.pad_token is None else tokenizer.pad_token
     base_model = AutoModelForCausalLM.from_pretrained(
+        "TinyLlaMA/TinyLlaMA-1.1B-Chat-v1.0",
         torch_dtype=torch.float32,
     ).to(device)