Update README.md
Browse files
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"
|