Update inference.py

Tencent · Jun 14, 2024 · 4398a5c · 4398a5c
1 parent 442ca2b
commit 4398a5c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/hydit/inference.py b/hydit/inference.py
@@ -202,6 +202,7 @@ def __init__(self, args, models_root_path):
 
         self.infer_mode = self.args.infer_mode
         if self.infer_mode in ['fa', 'torch']:
+            model_dir = self.root / "model"
             model_path = model_dir / f"pytorch_model_{self.args.load_key}.pt"
             if not model_path.exists():
                 raise ValueError(f"model_path not exists: {model_path}")