From 9a124f0ff09ecaa6dfc3ff2ad9d7234ce762aa86 Mon Sep 17 00:00:00 2001 From: xffxff <1247714429@qq.com> Date: Wed, 18 Dec 2024 10:50:28 +0800 Subject: [PATCH] Revert "update readme" This reverts commit 483af3d13c2ac5da0b47edf2d7fbfac9ab10e530. --- README.md | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 4c20f59..5e8d020 100644 --- a/README.md +++ b/README.md @@ -46,11 +46,10 @@ from PIL import Image from transformers import AutoModelForCausalLM, AutoProcessor model_id_or_path = "rhymes-ai/Aria" -revision = "4844f0b5ff678e768236889df5accbe4967ec845" -model = AutoModelForCausalLM.from_pretrained(model_id_or_path, revision=revision, device_map="auto", torch_dtype=torch.bfloat16, trust_remote_code=True) +model = AutoModelForCausalLM.from_pretrained(model_id_or_path, device_map="auto", torch_dtype=torch.bfloat16, trust_remote_code=True) -processor = AutoProcessor.from_pretrained(model_id_or_path, revision=revision, trust_remote_code=True) +processor = AutoProcessor.from_pretrained(model_id_or_path, trust_remote_code=True) image_path = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png"