Spaces:
Running
on
Zero
Running
on
Zero
wenxiang guo
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -98,7 +98,7 @@ class GenSamples:
|
|
| 98 |
|
| 99 |
return record_dicts
|
| 100 |
|
| 101 |
-
|
| 102 |
def infer(ori_prompt, ddim_steps, scale, seed):
|
| 103 |
# np.random.seed(seed)
|
| 104 |
# torch.manual_seed(seed)
|
|
|
|
| 98 |
|
| 99 |
return record_dicts
|
| 100 |
|
| 101 |
+
@spaces.GPU(enable_queue=True)
|
| 102 |
def infer(ori_prompt, ddim_steps, scale, seed):
|
| 103 |
# np.random.seed(seed)
|
| 104 |
# torch.manual_seed(seed)
|