Update app.py

app.py

@@ -233,12 +233,17 @@ import multiprocessing
 
 # 专门为子进程准备的常驻死循环
 def dagger_worker_process(worker_idx):
-    # 强制当前进程只使用 1 个线程，防止多进程抢占死锁
+    # 🚀 核心修复：在子进程内部重新 import，确保它一定能找到环境！
+    import os
+    import time
+    import torch
+    
+    # 强制当前进程只使用 1 个线程，防止多进程抢占 CPU 导致死锁
     torch.set_num_threads(1)
     os.environ["OMP_NUM_THREADS"] = "1"
     
     print(f"👷 [Worker {worker_idx}] 启动中，正在加载专属常驻模型...")
-    # 子进程独立加载模型，防止多进程共享 PyTorch 对象的内存冲突
+    # 子进程独立加载模型，防止多进程共享 PyTorch 对象的显存/内存冲突
     student_bots = {i: load_model_student(i, STUDENT_MODEL) for i in range(3)}
     teacher_bots = {i: load_model_teacher(i, TEACHER_MODEL) for i in range(3)}
     encoder = DAggerEncoder(chunk_size=2048)