Skip to content

Commit bb5a9b6

Browse files
authored
fix moondream2
1 parent 99706c6 commit bb5a9b6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/module_process_images.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,7 @@ def initialize_model_and_tokenizer(self):
152152

153153
model = AutoModelForCausalLM.from_pretrained(model_id,
154154
trust_remote_code=True,
155-
revision="2024-05-20",
155+
revision="2024-07-23",
156156
torch_dtype=torch.float16,
157157
cache_dir=cache_dir,
158158
low_cpu_mem_usage=True).to(self.device)

0 commit comments

Comments
 (0)