From b35d4dd7951efd2816ef83d0a2be344e95541bc5 Mon Sep 17 00:00:00 2001 From: systemd Date: Mon, 2 Mar 2026 01:30:47 +0000 Subject: [PATCH] Adding `transformers` as the library name (#4) - Adding `transformers` as the library name (37fdde0d1f220c0286925927d2fe6718b22715de) Co-authored-by: Aritra Roy Gosthipaty (batch 1/1) --- README.md | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index e003aac..e520098 100644 --- a/README.md +++ b/README.md @@ -1,6 +1,7 @@ --- license: apache-2.0 pipeline_tag: image-text-to-text +library_name: transformers --- Chat @@ -93,7 +94,7 @@ model = Qwen3VLForConditionalGeneration.from_pretrained( # device_map="auto", # ) -processor = AutoProcessor.from_pretrained("Qwen/Qwen/Qwen3-VL-8B-Thinking") +processor = AutoProcessor.from_pretrained("Qwen/Qwen3-VL-8B-Thinking") messages = [ { @@ -116,6 +117,7 @@ inputs = processor.apply_chat_template( return_dict=True, return_tensors="pt" ) +inputs = inputs.to(model.device) # Inference: Generation of the output generated_ids = model.generate(**inputs, max_new_tokens=128)