Spaces:
Runtime error
Runtime error
Update src/generate_utils/generate.py
Browse files
src/generate_utils/generate.py
CHANGED
|
@@ -27,7 +27,7 @@ class PressureGenerator:
|
|
| 27 |
|
| 28 |
def __init__(self,
|
| 29 |
ckpt_dir="src/generate_utils/lib/ckpt/pressurepose_20251222_180032",
|
| 30 |
-
smpl_model_dir="smpl_models",
|
| 31 |
device="cpu"):
|
| 32 |
"""
|
| 33 |
初始化生成器:加载配置、权重和 SMPL 模型。
|
|
|
|
| 27 |
|
| 28 |
def __init__(self,
|
| 29 |
ckpt_dir="src/generate_utils/lib/ckpt/pressurepose_20251222_180032",
|
| 30 |
+
smpl_model_dir="src/smpl_models",
|
| 31 |
device="cpu"):
|
| 32 |
"""
|
| 33 |
初始化生成器:加载配置、权重和 SMPL 模型。
|