CodeGoat24 commited on
Commit
dcb4314
·
verified ·
1 Parent(s): 1814c61

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -66,7 +66,7 @@ import os
66
 
67
 
68
  warnings.filterwarnings("ignore")
69
- pretrained = "CodeGoat24/UnifiedReward"
70
  model_name = "llava_qwen"
71
  device = "cuda"
72
  device_map = "auto"
 
66
 
67
 
68
  warnings.filterwarnings("ignore")
69
+ pretrained = "CodeGoat24/UnifiedReward-7b"
70
  model_name = "llava_qwen"
71
  device = "cuda"
72
  device_map = "auto"