PommesPeter commited on
Commit
5f08391
1 Parent(s): 709d8be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -171,7 +171,7 @@ def load_model(args, master_port, rank):
171
 
172
  @spaces.GPU
173
  @torch.no_grad()
174
- def model_main(args, master_port, rank, request_queue, response_queue, text_encoder, tokenizer, vae, model):
175
  dtype = {"bf16": torch.bfloat16, "fp16": torch.float16, "fp32": torch.float32}[
176
  args.precision
177
  ]
 
171
 
172
  @spaces.GPU
173
  @torch.no_grad()
174
+ def model_main(args, master_port, rank, request_queue, response_queue):
175
  dtype = {"bf16": torch.bfloat16, "fp16": torch.float16, "fp32": torch.float32}[
176
  args.precision
177
  ]