fix model bug
Browse files
app.py
CHANGED
|
@@ -295,8 +295,9 @@ def run_edit(
|
|
| 295 |
# --- 关键修改:模型上 GPU ---
|
| 296 |
# model_cpu.to(device) 是原位操作!会修改全局缓存!
|
| 297 |
# 所以必须在 finally 里搬回去,或者在这里使用深拷贝(深拷贝太慢,建议搬回去)
|
|
|
|
| 298 |
model = model_cpu.to(device).eval()
|
| 299 |
-
|
| 300 |
# --- 数据预处理 ---
|
| 301 |
seed = int(seed)
|
| 302 |
torch.manual_seed(seed)
|
|
|
|
| 295 |
# --- 关键修改:模型上 GPU ---
|
| 296 |
# model_cpu.to(device) 是原位操作!会修改全局缓存!
|
| 297 |
# 所以必须在 finally 里搬回去,或者在这里使用深拷贝(深拷贝太慢,建议搬回去)
|
| 298 |
+
|
| 299 |
model = model_cpu.to(device).eval()
|
| 300 |
+
logger.info("Moving model to GPU for inference...")
|
| 301 |
# --- 数据预处理 ---
|
| 302 |
seed = int(seed)
|
| 303 |
torch.manual_seed(seed)
|