fix moondream2

BBC-Esq · Jul 27, 2024 · bb5a9b6 · bb5a9b6
1 parent 99706c6
commit bb5a9b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/module_process_images.py b/src/module_process_images.py
@@ -152,7 +152,7 @@ def initialize_model_and_tokenizer(self):
 
         model = AutoModelForCausalLM.from_pretrained(model_id, 
                                                      trust_remote_code=True, 
-                                                     revision="2024-05-20",
+                                                     revision="2024-07-23",
                                                      torch_dtype=torch.float16,
                                                      cache_dir=cache_dir,
                                                      low_cpu_mem_usage=True).to(self.device)