Spaces:

gijs
/

phi4_mm

Runtime error

Gijs Wijngaard commited on Aug 26

Commit

f931e52

1 Parent(s): 3d568e9

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,6 +17,7 @@ model = AutoModelForCausalLM.from_pretrained(
     MODEL_ID,
     device_map="cuda" if torch.cuda.is_available() else "cpu",
     torch_dtype="auto",
 )
 model.load_adapter(MODEL_ID, adapter_name="speech", device_map="cuda" if torch.cuda.is_available() else "cpu", adapter_kwargs={"subfolder": 'speech-lora'})

     MODEL_ID,
     device_map="cuda" if torch.cuda.is_available() else "cpu",
     torch_dtype="auto",
+    trust_remote_code=True
 )
 model.load_adapter(MODEL_ID, adapter_name="speech", device_map="cuda" if torch.cuda.is_available() else "cpu", adapter_kwargs={"subfolder": 'speech-lora'})

more