Skip to content

Commit 9a124f0

Browse files
committed
Revert "update readme"
This reverts commit 483af3d.
1 parent f4c840e commit 9a124f0

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

README.md

+2-3
Original file line numberDiff line numberDiff line change
@@ -46,11 +46,10 @@ from PIL import Image
4646
from transformers import AutoModelForCausalLM, AutoProcessor
4747

4848
model_id_or_path = "rhymes-ai/Aria"
49-
revision = "4844f0b5ff678e768236889df5accbe4967ec845"
5049

51-
model = AutoModelForCausalLM.from_pretrained(model_id_or_path, revision=revision, device_map="auto", torch_dtype=torch.bfloat16, trust_remote_code=True)
50+
model = AutoModelForCausalLM.from_pretrained(model_id_or_path, device_map="auto", torch_dtype=torch.bfloat16, trust_remote_code=True)
5251

53-
processor = AutoProcessor.from_pretrained(model_id_or_path, revision=revision, trust_remote_code=True)
52+
processor = AutoProcessor.from_pretrained(model_id_or_path, trust_remote_code=True)
5453

5554
image_path = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png"
5655

0 commit comments

Comments
 (0)