lmms-lab
/

Qwen2-VL-7B-GRPO-8k

Model card Files Files and versions Community

kcz358 commited on 3 days ago

Commit

a7f7d7e

·

verified ·

1 Parent(s): 6e1c2ab

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -36,7 +36,7 @@ SYSTEM_PROMPT = (
 # default: Load the model on the available device(s)
 model = Qwen2VLForConditionalGeneration.from_pretrained(
-    "lmms-lab/Qwen2-VL-7B-GRPO-8k", torch_dtype="auto", device_map="auto"
 )
 # default processer

 # default: Load the model on the available device(s)
 model = Qwen2VLForConditionalGeneration.from_pretrained(
+    "lmms-lab/Qwen2-VL-7B-GRPO-8k", torch_dtype="auto", device_map="cuda"
 )
 # default processer