karimbenharrak commited on
Commit
b6cc2ab
1 Parent(s): 9d60b6e

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +0 -2
handler.py CHANGED
@@ -66,7 +66,6 @@ class EndpointHandler():
66
  guidance_scale=8.0,
67
  num_inference_steps=100,
68
  strength=0.2,
69
- generator=generator,
70
  output_type="latent", # let's keep in latent to save some VRAM
71
  ).images[0]
72
 
@@ -78,7 +77,6 @@ class EndpointHandler():
78
  guidance_scale=8.0,
79
  num_inference_steps=100,
80
  strength=0.2,
81
- generator=generator,
82
  ).images[0]
83
 
84
  # return first generate PIL image
 
66
  guidance_scale=8.0,
67
  num_inference_steps=100,
68
  strength=0.2,
 
69
  output_type="latent", # let's keep in latent to save some VRAM
70
  ).images[0]
71
 
 
77
  guidance_scale=8.0,
78
  num_inference_steps=100,
79
  strength=0.2,
 
80
  ).images[0]
81
 
82
  # return first generate PIL image