sd24 commited on
Commit
f344ae3
·
verified ·
1 Parent(s): 35055b7

Upload folder using huggingface_hub

Browse files
README.md CHANGED
@@ -52,22 +52,14 @@ import requests
52
  repo_id = "<repo_id>"
53
  model = AutoModelForImageTextToText.from_pretrained(
54
  repo_id, trust_remote_code=True,
55
- ).cuda() # fla's RMSNorm uses Triton kernels that only run on CUDA tensors.
56
  processor = AutoProcessor.from_pretrained(repo_id, trust_remote_code=True)
57
 
58
  url = "https://llava-vl.github.io/static/images/view.jpg"
59
  image = Image.open(requests.get(url, stream=True).raw)
 
 
60
 
61
- # PanoVLM's chat template wraps string content, so put the <|image|> placeholder
62
- # inline in the message text (the processor expands it into the image tokens).
63
- messages = [{"role": "user", "content": "<|image|>Describe this image."}]
64
- prompt = processor.tokenizer.apply_chat_template(
65
- messages, add_generation_prompt=True, tokenize=False,
66
- )
67
- inputs = processor(text=prompt, images=image, return_tensors="pt").to(model.device)
68
-
69
- # use_cache=False: HF's generate() passes a DynamicCache that fla indexes as a
70
- # list, which the new transformers API no longer supports.
71
  out = model.generate(**inputs, max_new_tokens=512, temperature=1.0, top_k=10,
72
  top_p=1.0, do_sample=True, use_cache=False)
73
  print(processor.decode(out[0], skip_special_tokens=True))
 
52
  repo_id = "<repo_id>"
53
  model = AutoModelForImageTextToText.from_pretrained(
54
  repo_id, trust_remote_code=True,
55
+ ).cuda()
56
  processor = AutoProcessor.from_pretrained(repo_id, trust_remote_code=True)
57
 
58
  url = "https://llava-vl.github.io/static/images/view.jpg"
59
  image = Image.open(requests.get(url, stream=True).raw)
60
+ inputs = processor(text="<|image|>An image of", images=image,
61
+ return_tensors="pt").to(model.device)
62
 
 
 
 
 
 
 
 
 
 
 
63
  out = model.generate(**inputs, max_new_tokens=512, temperature=1.0, top_k=10,
64
  top_p=1.0, do_sample=True, use_cache=False)
65
  print(processor.decode(out[0], skip_special_tokens=True))
model-00001-of-00001.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f731da549b2e1f484b6e3f09c5465f54c8990e90cb9410c405d641ec2a8f42f4
3
  size 2052770433
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c545658c0c55d652b878c0814f69d802960aad2d1f2629951370df1b480d226
3
  size 2052770433
sharded/shard-00001-model-00001-of-00001.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bbf762a31e4d2ae83c0df1131f2798309df8b1c6f97fd31c0f365c94f5d948a3
3
  size 2052858480
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0b31434754ac80fd0884c829fcb39706781194a8914fb332ae51def8efe4f539
3
  size 2052858480