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 |
(
|