Spaces:

rlawjdghek
/

StableVITON

Running on Zero

rlawjdghek commited on Apr 27

Commit

700defa

•

1 Parent(s): 4175830

add spaces GPU

Files changed (1) hide show

app.py CHANGED Viewed

@@ -124,7 +124,7 @@ def stable_viton_model_hd2(
     start_code = model2.q_sample(z, ts)
     torch.cuda.empty_cache()
     output, _, _ = sampler2.sample(
-        n_steps,
         bs,
         (4, IMG_H//8, IMG_W//8),
         cond,

     start_code = model2.q_sample(z, ts)
     torch.cuda.empty_cache()
     output, _, _ = sampler2.sample(
+        n_steps,
         bs,
         (4, IMG_H//8, IMG_W//8),
         cond,