karimbenharrak
commited on
Commit
•
b6cc2ab
1
Parent(s):
9d60b6e
Update handler.py
Browse files- 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
|