Spaces:

comrender
/

fluxhdupscaler

Runtime error

comrender commited on Jul 28

Commit

40a4e69

verified ·

1 Parent(s): 7f1edc9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -50,7 +50,8 @@ print("📥 Loading Florence-2 model...")
 florence_model = AutoModelForCausalLM.from_pretrained(
     "microsoft/Florence-2-large",
     torch_dtype=torch.float16,
-    trust_remote_code=True
 ).to(device)
 florence_processor = AutoProcessor.from_pretrained(
     "microsoft/Florence-2-large",

 florence_model = AutoModelForCausalLM.from_pretrained(
     "microsoft/Florence-2-large",
     torch_dtype=torch.float16,
+    trust_remote_code=True,
+    attn_implementation="eager"  # Fix for SDPA compatibility issue
 ).to(device)
 florence_processor = AutoProcessor.from_pretrained(
     "microsoft/Florence-2-large",