Spaces:
Runtime error
Runtime error
update rl model
#2
by
zzjzzj
- opened
app.py
CHANGED
|
@@ -53,10 +53,15 @@ models = {
|
|
| 53 |
trust_remote_code=True,
|
| 54 |
torch_dtype="auto",
|
| 55 |
device_map="auto").eval(),
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
}
|
| 57 |
|
| 58 |
processors = {
|
| 59 |
"Fancy-MLLM/R1-Onevision-7B": AutoProcessor.from_pretrained("Fancy-MLLM/R1-Onevision-7B", trust_remote_code=True),
|
|
|
|
| 60 |
}
|
| 61 |
|
| 62 |
DESCRIPTION = "[π¦ Fancy-MLLM/R1-Onevision-7B Demo]"
|
|
|
|
| 53 |
trust_remote_code=True,
|
| 54 |
torch_dtype="auto",
|
| 55 |
device_map="auto").eval(),
|
| 56 |
+
"Fancy-MLLM/R1-Onevision-7B-RL": Qwen2_5_VLForConditionalGeneration.from_pretrained("Fancy-MLLM/R1-Onevision-7B-RL",
|
| 57 |
+
trust_remote_code=True,
|
| 58 |
+
torch_dtype="auto",
|
| 59 |
+
device_map="auto").eval(),
|
| 60 |
}
|
| 61 |
|
| 62 |
processors = {
|
| 63 |
"Fancy-MLLM/R1-Onevision-7B": AutoProcessor.from_pretrained("Fancy-MLLM/R1-Onevision-7B", trust_remote_code=True),
|
| 64 |
+
"Fancy-MLLM/R1-Onevision-7B-RL": AutoProcessor.from_pretrained("Fancy-MLLM/R1-Onevision-7B-RL", trust_remote_code=True),
|
| 65 |
}
|
| 66 |
|
| 67 |
DESCRIPTION = "[π¦ Fancy-MLLM/R1-Onevision-7B Demo]"
|