yolozyk commited on
Commit
f5efd0f
·
verified ·
1 Parent(s): 8a85d97

Update src/generate_utils/generate.py

Browse files
Files changed (1) hide show
  1. src/generate_utils/generate.py +1 -1
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 模型。