alvarobartt HF staff commited on
Commit
87c555a
1 Parent(s): 253e509

Update handler.py

Browse files
Files changed (1) hide show
  1. 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]