Spaces:

tjw
/

gauge

Running on Zero

tjw commited on Nov 11

Commit

6261698

•

1 Parent(s): 90df72c

update

Files changed (1) hide show

aimodel.py CHANGED Viewed

@@ -18,8 +18,7 @@ class Florence:
             return
         self.model = (
             AutoModelForCausalLM.from_pretrained(
-                model_id, trust_remote_code=True, torch_dtype="auto",
-                trust_remote_code=True
             )
             .eval()
             .cuda()

             return
         self.model = (
             AutoModelForCausalLM.from_pretrained(
+                model_id, trust_remote_code=True, torch_dtype="auto"
             )
             .eval()
             .cuda()