Commit
•
87c555a
1
Parent(s):
253e509
Update handler.py
Browse files- handler.py +1 -3
handler.py
CHANGED
@@ -44,7 +44,6 @@ class EndpointHandler:
|
|
44 |
width = parameters.get("width", 1024)
|
45 |
height = parameters.get("height", 768)
|
46 |
guidance_scale = parameters.get("guidance_scale", 3.5)
|
47 |
-
lora_scale = parameters.get("lora_scale", 1.0)
|
48 |
|
49 |
# seed generator (seed cannot be provided as is but via a generator)
|
50 |
seed = parameters.get("seed", 0)
|
@@ -56,6 +55,5 @@ class EndpointHandler:
|
|
56 |
width=width,
|
57 |
guidance_scale=guidance_scale,
|
58 |
num_inference_steps=num_inference_steps,
|
59 |
-
lora_scale=lora_scale,
|
60 |
generator=generator,
|
61 |
-
).images[0]
|
|
|
44 |
width = parameters.get("width", 1024)
|
45 |
height = parameters.get("height", 768)
|
46 |
guidance_scale = parameters.get("guidance_scale", 3.5)
|
|
|
47 |
|
48 |
# seed generator (seed cannot be provided as is but via a generator)
|
49 |
seed = parameters.get("seed", 0)
|
|
|
55 |
width=width,
|
56 |
guidance_scale=guidance_scale,
|
57 |
num_inference_steps=num_inference_steps,
|
|
|
58 |
generator=generator,
|
59 |
+
).images[0]
|