Spaces:
Running
on
Zero
Running
on
Zero
guowenxiang
commited on
Commit
•
0fac373
1
Parent(s):
14db764
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)
|