Enferlain commited on
Commit
5d94273
·
verified ·
1 Parent(s): faec259

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -153,7 +153,7 @@ class ModelManager:
153
  self.vision_model = AutoModel.from_pretrained(
154
  base_vision_model_name, torch_dtype=torch.float16 if DEVICE == "cuda" else torch.float32
155
  ).to(DEVICE).eval()
156
- else: # For SigLIP or other non-8bit models
157
  self.hf_processor = AutoProcessor.from_pretrained(base_vision_model_name)
158
  self.vision_model = AutoModel.from_pretrained(
159
  base_vision_model_name,
 
153
  self.vision_model = AutoModel.from_pretrained(
154
  base_vision_model_name, torch_dtype=torch.float16 if DEVICE == "cuda" else torch.float32
155
  ).to(DEVICE).eval()
156
+ else:
157
  self.hf_processor = AutoProcessor.from_pretrained(base_vision_model_name)
158
  self.vision_model = AutoModel.from_pretrained(
159
  base_vision_model_name,