q-future
/

one-align

Zero-Shot Image Classification

feature-extraction

Model card Files Files and versions

teowu commited on Jan 19, 2024

Commit

b87e7e0

·

verified ·

1 Parent(s): 728ace9

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -12,7 +12,7 @@ import requests
 import torch
 from transformers import AutoModelForCausalLM
-model = AutoModelForCausalLM.from_pretrained("q-future/one-align", trust_remote_code=True,
                                              torch_dtype=torch.float16, device_map="auto")
 from PIL import Image

 import torch
 from transformers import AutoModelForCausalLM
+model = AutoModelForCausalLM.from_pretrained("q-future/one-align", trust_remote_code=True, attn_implementation="eager",
                                              torch_dtype=torch.float16, device_map="auto")
 from PIL import Image