CocoBro commited on
Commit
d52557e
·
1 Parent(s): f3f0643

fix model bug

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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)