Commit
•
3acc825
1
Parent(s):
93e1d77
Fix argument
Browse files- gradio_demo.py +0 -2
gradio_demo.py
CHANGED
|
@@ -182,7 +182,6 @@ def stage2_process(
|
|
| 182 |
|
| 183 |
samples = restore(
|
| 184 |
model,
|
| 185 |
-
captions,
|
| 186 |
edm_steps,
|
| 187 |
s_stage1,
|
| 188 |
s_churn,
|
|
@@ -239,7 +238,6 @@ def stage2_process(
|
|
| 239 |
@spaces.GPU(duration=600)
|
| 240 |
def restore(
|
| 241 |
model,
|
| 242 |
-
captions,
|
| 243 |
edm_steps,
|
| 244 |
s_stage1,
|
| 245 |
s_churn,
|
|
|
|
| 182 |
|
| 183 |
samples = restore(
|
| 184 |
model,
|
|
|
|
| 185 |
edm_steps,
|
| 186 |
s_stage1,
|
| 187 |
s_churn,
|
|
|
|
| 238 |
@spaces.GPU(duration=600)
|
| 239 |
def restore(
|
| 240 |
model,
|
|
|
|
| 241 |
edm_steps,
|
| 242 |
s_stage1,
|
| 243 |
s_churn,
|