update rl model

#2
by zzjzzj - opened
Files changed (1) hide show
  1. app.py +5 -0
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]"