amildravid4292 commited on
Commit
01aa716
·
verified ·
1 Parent(s): e615641

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -42,6 +42,7 @@ def sample_model():
42
 
43
  @torch.no_grad()
44
  def inference(prompt, negative_prompt, guidance_scale, ddim_steps, seed):
 
45
  global device
46
  global generator
47
  global unet
 
42
 
43
  @torch.no_grad()
44
  def inference(prompt, negative_prompt, guidance_scale, ddim_steps, seed):
45
+ global network
46
  global device
47
  global generator
48
  global unet