Update README.md
Browse files
README.md
CHANGED
@@ -36,7 +36,7 @@ SYSTEM_PROMPT = (
|
|
36 |
|
37 |
# default: Load the model on the available device(s)
|
38 |
model = Qwen2VLForConditionalGeneration.from_pretrained(
|
39 |
-
"
|
40 |
)
|
41 |
|
42 |
# default processer
|
|
|
36 |
|
37 |
# default: Load the model on the available device(s)
|
38 |
model = Qwen2VLForConditionalGeneration.from_pretrained(
|
39 |
+
"lmms-lab/Qwen2-VL-7B-GRPO-8k", torch_dtype="auto", device_map="auto"
|
40 |
)
|
41 |
|
42 |
# default processer
|