LanguageMachines
/

blip2-flan-t5-xxl

image-captioning

visual-question-answering

Model card Files Files and versions

sarang-shrivastava commited on Jun 28, 2023

Commit

b233fe4

·

1 Parent(s): 6f1c35a

Update handler

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -24,7 +24,7 @@ class EndpointHandler():
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
-        self.model.to(self.device)
         logging.info('Model moved to device-' + self.device)

         self.device = "cuda" if torch.cuda.is_available() else "cpu"
+        # self.model.to(self.device)
         logging.info('Model moved to device-' + self.device)