fffiloni commited on
Commit
ccb508e
·
1 Parent(s): b486fa5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def load_model(custom_model):
24
  pipe.to(device)
25
  return "Model loaded!"
26
 
27
- def infer (prompt, inf_steps, guidance_scale, seed, lora_weigth, progress=gr.Progress(track_tqdm=True)):
28
  generator = torch.Generator(device="cuda").manual_seed(seed)
29
  image = pipe(
30
  prompt=prompt,
 
24
  pipe.to(device)
25
  return "Model loaded!"
26
 
27
+ def infer (prompt, inf_steps, guidance_scale, seed, lora_weight, progress=gr.Progress(track_tqdm=True)):
28
  generator = torch.Generator(device="cuda").manual_seed(seed)
29
  image = pipe(
30
  prompt=prompt,