llava-hf
/

llava-v1.6-34b-hf

Image-Text-to-Text

Inference Endpoints

Model card Files Files and versions Community

RaushanTurganbay HF staff commited on Aug 16

Commit

66b6feb

•

1 Parent(s): 982f7ce

update processor kwargs

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -65,7 +65,7 @@ conversation = [
 ]
 prompt = processor.apply_chat_template(conversation, add_generation_prompt=True)
-inputs = processor(prompt, image, return_tensors="pt").to("cuda:0")
 # autoregressively complete prompt
 output = model.generate(**inputs, max_new_tokens=100)

 ]
 prompt = processor.apply_chat_template(conversation, add_generation_prompt=True)
+inputs = processor(images=image, text=prompt, return_tensors="pt").to("cuda:0")
 # autoregressively complete prompt
 output = model.generate(**inputs, max_new_tokens=100)