q-future
/

one-align

Zero-Shot Image Classification

feature-extraction

Model card Files Files and versions Community

teowu commited on Jan 3

Commit

0033227

•

1 Parent(s): 7b864d8

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -12,7 +12,7 @@ import torch
 from transformers import AutoModelForCausalLM
 model = AutoModelForCausalLM.from_pretrained("q-future/one-align", trust_remote_code=True,
-                                             torch_dtype=torch.float16, device_map={"":"cuda:0"})
 from PIL import Image
 url = "https://raw.githubusercontent.com/Q-Future/Q-Align/main/fig/singapore_flyer.jpg"

 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
 url = "https://raw.githubusercontent.com/Q-Future/Q-Align/main/fig/singapore_flyer.jpg"