ChirathD
/

Blip-2-test-1

Visual Question Answering

Inference Endpoints

8-bit precision

Model card Files Files and versions Community

ChirathD commited on Dec 6, 2023

Commit

d703739

•

1 Parent(s): 9e4dfd9

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -20,7 +20,7 @@ class EndpointHandler():
         print("####### Start Deploying #####")
         self.processor = Blip2Processor.from_pretrained("ChirathD/Blip-2-test-1")
         self.model = Blip2ForConditionalGeneration.from_pretrained("ChirathD/Blip-2-test-1")
-        self.model.eval()
         # self.model = self.model.to(device)

         print("####### Start Deploying #####")
         self.processor = Blip2Processor.from_pretrained("ChirathD/Blip-2-test-1")
         self.model = Blip2ForConditionalGeneration.from_pretrained("ChirathD/Blip-2-test-1")
+        # self.model.eval()
         # self.model = self.model.to(device)