arjunanand13
/

florence-enphaseall2-25e

Image-Text-to-Text

Model card Files Files and versions

arjunanand13 commited on Oct 14, 2024

Commit

9525474

·

verified ·

1 Parent(s): f84923c

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -22,7 +22,7 @@ class EndpointHandler:
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         print(f"Using device: {self.device}")
-        self.model_name = "arjunanand13/florence-enphaseall2-30e"
         self.model = AutoModelForCausalLM.from_pretrained(
             self.model_name,
             trust_remote_code=True,

         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         print(f"Using device: {self.device}")
+        self.model_name = "arjunanand13/florence-enphaseall2-25e"
         self.model = AutoModelForCausalLM.from_pretrained(
             self.model_name,
             trust_remote_code=True,