Update app.py
Browse files
app.py
CHANGED
@@ -18,8 +18,8 @@ def setup_environment():
|
|
18 |
setup_environment()
|
19 |
|
20 |
# 在环境设置后再导入项目内的模块
|
21 |
-
from
|
22 |
-
from
|
23 |
|
24 |
# 模型ID
|
25 |
model_id = "Skywork/Skywork-o1-Open-PRM-Qwen-2.5-1.5B"
|
|
|
18 |
setup_environment()
|
19 |
|
20 |
# 在环境设置后再导入项目内的模块
|
21 |
+
from model_utils.prm_model import PRM_MODEL
|
22 |
+
from model_utils.io_utils import prepare_input, prepare_batch_input_for_model, derive_step_rewards
|
23 |
|
24 |
# 模型ID
|
25 |
model_id = "Skywork/Skywork-o1-Open-PRM-Qwen-2.5-1.5B"
|