Use main branches

app.py

@@ -6,12 +6,11 @@ from transformers.models.fuyu.image_processing_fuyu import FuyuImageProcessor
 from PIL import Image
 
 model_id = "adept/fuyu-8b"
-revision = "refs/pr/3"
 dtype = torch.bfloat16
 device = "cuda"
 
-tokenizer = AutoTokenizer.from_pretrained(model_id, revision=revision)
-model = FuyuForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=dtype, revision=revision)
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = FuyuForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=dtype)
 processor = FuyuProcessor(image_processor=FuyuImageProcessor(), tokenizer=tokenizer)
 
 caption_prompt = "Generate a coco-style caption.\\n"

requirements.txt

@@ -1,3 +1,3 @@
-git+https://github.com/huggingface/transformers.git@add_fuyu_model
+git+https://github.com/huggingface/transformers.git
 accelerate
 torch==2.0.1