Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -162,7 +162,7 @@ def model_main(args, master_port, rank):
|
|
| 162 |
|
| 163 |
|
| 164 |
@torch.no_grad()
|
| 165 |
-
def inference(args, infer_args, text_encoder, tokenizer, vae, model)
|
| 166 |
with torch.autocast("cuda", dtype):
|
| 167 |
while True:
|
| 168 |
(
|
|
|
|
| 162 |
|
| 163 |
|
| 164 |
@torch.no_grad()
|
| 165 |
+
def inference(args, infer_args, text_encoder, tokenizer, vae, model):
|
| 166 |
with torch.autocast("cuda", dtype):
|
| 167 |
while True:
|
| 168 |
(
|