lightonai
/

MonoQwen2-VL-v0.1

Model card Files Files and versions Community

uminaty commited on Oct 25, 2024

Commit

7923b22

·

verified ·

1 Parent(s): d902199

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -43,7 +43,7 @@ messages = [
 # Apply chat template and tokenize
 text = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
-inputs = processor(text=text, images=image, return_tensors="pt").to("cuda:1")
 # Run inference to obtain logits
 with torch.no_grad():

 # Apply chat template and tokenize
 text = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
+inputs = processor(text=text, images=image, return_tensors="pt")
 # Run inference to obtain logits
 with torch.no_grad():