megalado commited on
Commit
f6303c9
·
1 Parent(s): f5cd424

run sampler in --unconstrained mode (no dataset needed)

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -20,12 +20,14 @@ def generate_motion(prompt: str) -> str:
20
  "python", "-m", "motion_diffusion_model.sample.generate",
21
  "--model_path", str(CKPT_PATH),
22
  "--text_prompt", prompt,
23
- "--cuda", "-1", # -1 => CPU
24
  "--guidance_param", "7.5",
25
  "--output_dir", str(out_dir),
26
- "--num_samples", "1"
 
27
  ]
28
 
 
29
  env = os.environ.copy()
30
  root = Path(__file__).parent
31
  repo = root / "motion_diffusion_model"
 
20
  "python", "-m", "motion_diffusion_model.sample.generate",
21
  "--model_path", str(CKPT_PATH),
22
  "--text_prompt", prompt,
23
+ "--cuda", "-1",
24
  "--guidance_param", "7.5",
25
  "--output_dir", str(out_dir),
26
+ "--num_samples", "1",
27
+ "--unconstrained" # ← add this line
28
  ]
29
 
30
+
31
  env = os.environ.copy()
32
  root = Path(__file__).parent
33
  repo = root / "motion_diffusion_model"