We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 99706c6 commit bb5a9b6Copy full SHA for bb5a9b6
src/module_process_images.py
@@ -152,7 +152,7 @@ def initialize_model_and_tokenizer(self):
152
153
model = AutoModelForCausalLM.from_pretrained(model_id,
154
trust_remote_code=True,
155
- revision="2024-05-20",
+ revision="2024-07-23",
156
torch_dtype=torch.float16,
157
cache_dir=cache_dir,
158
low_cpu_mem_usage=True).to(self.device)
0 commit comments