Spaces:

yasserrmd
/

dlp_assistant

Sleeping

yasserrmd commited on 24 days ago

Commit

db11727

verified ·

1 Parent(s): 023eddc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import torch
 model_name = "yasserrmd/SmolLM2-135M-synthetic-dlp"
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
-    torch_dtype=torch.bfloat16
     device_map ="cuda"
 )
 tokenizer = AutoTokenizer.from_pretrained(model_name)

 model_name = "yasserrmd/SmolLM2-135M-synthetic-dlp"
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
+    torch_dtype=torch.bfloat16,
     device_map ="cuda"
 )
 tokenizer = AutoTokenizer.from_pretrained(model_name)