Spaces:

Daemontatox
/

Mawared-Support-Assistant

Running

Daemontatox commited on Jan 12

Commit

ec56115

verified ·

1 Parent(s): 72ccc50

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -145,7 +145,7 @@ model_id = "unsloth/phi-4"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(
-        MODEL_ID,
         torch_dtype=torch.float16,
         device_map="cuda",
         attn_implementation="flash_attention_2",

 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(
+        model_id,
         torch_dtype=torch.float16,
         device_map="cuda",
         attn_implementation="flash_attention_2",