Spaces:

robinhad
/

llama-4-Scout-17B-16E-Instruct

Runtime error

robinhad commited on Apr 5

Commit

db740ce

verified ·

1 Parent(s): 04e4e52

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -63,7 +63,7 @@ processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
 ).to("cuda").eval()"""
 model = Llama4ForConditionalGeneration.from_pretrained(
-    model_id,
     attn_implementation="flex_attention",
     device_map="auto",
     torch_dtype=torch.bfloat16,

 ).to("cuda").eval()"""
 model = Llama4ForConditionalGeneration.from_pretrained(
+    MODEL_ID,
     attn_implementation="flex_attention",
     device_map="auto",
     torch_dtype=torch.bfloat16,