ganeshnanduru commited on
Commit
4496a06
·
verified ·
1 Parent(s): 135c613

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -79,12 +79,14 @@ from transformers import Zaya1VLForConditionalGeneration, Zaya1VLProcessor
79
  import torch
80
  from PIL import Image
81
  from qwen_vl_utils import process_vision_info
 
82
 
83
  device = "cuda"
84
  processor = Zaya1VLProcessor.from_pretrained("Zyphra/ZAYA1-VL-8B", temporal_patch_size=1)
85
  model = Zaya1VLForConditionalGeneration.from_pretrained("Zyphra/ZAYA1-VL-8B", device_map=device, torch_dtype=torch.bfloat16, attn_implementation="flash_attention_2")
86
 
87
- image = Image.open("http://images.cocodataset.org/val2017/000000039769.jpg")
 
88
  question = "What do you see in the image? Give us some detail."
89
  num_img_tokens = 8000
90
 
 
79
  import torch
80
  from PIL import Image
81
  from qwen_vl_utils import process_vision_info
82
+ import requests
83
 
84
  device = "cuda"
85
  processor = Zaya1VLProcessor.from_pretrained("Zyphra/ZAYA1-VL-8B", temporal_patch_size=1)
86
  model = Zaya1VLForConditionalGeneration.from_pretrained("Zyphra/ZAYA1-VL-8B", device_map=device, torch_dtype=torch.bfloat16, attn_implementation="flash_attention_2")
87
 
88
+ url = "http://images.cocodataset.org/val2017/000000039769.jpg"
89
+ image = Image.open(requests.get(url, stream=True).raw)
90
  question = "What do you see in the image? Give us some detail."
91
  num_img_tokens = 8000
92