Spaces:

sitammeur
/

TextSnap

Runtime error

sitammeur commited on 21 days ago

Commit

6b8d652

verified ·

1 Parent(s): cd32385

Update src/florence/model.py

Files changed (1) hide show

src/florence/model.py CHANGED Viewed

@@ -23,7 +23,7 @@ subprocess.run(
 model_id = "microsoft/Florence-2-large-ft"
 try:
     model = (
-        AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True, from_tf=True)
         .to("cuda")
         .eval()
     )

 model_id = "microsoft/Florence-2-large-ft"
 try:
     model = (
+        AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
         .to("cuda")
         .eval()
     )