Spaces:
Running
on
Zero
Running
on
Zero
Commit
β’
700defa
1
Parent(s):
4175830
add spaces GPU
Browse files
app.py
CHANGED
@@ -124,7 +124,7 @@ def stable_viton_model_hd2(
|
|
124 |
start_code = model2.q_sample(z, ts)
|
125 |
torch.cuda.empty_cache()
|
126 |
output, _, _ = sampler2.sample(
|
127 |
-
n_steps,
|
128 |
bs,
|
129 |
(4, IMG_H//8, IMG_W//8),
|
130 |
cond,
|
|
|
124 |
start_code = model2.q_sample(z, ts)
|
125 |
torch.cuda.empty_cache()
|
126 |
output, _, _ = sampler2.sample(
|
127 |
+
n_steps,
|
128 |
bs,
|
129 |
(4, IMG_H//8, IMG_W//8),
|
130 |
cond,
|