Update src/pipeline.py
Browse files- src/pipeline.py +1 -1
src/pipeline.py
CHANGED
@@ -39,5 +39,5 @@ def load_pipeline() -> Pipeline:
|
|
39 |
def infer(request: TextToImageRequest, pipeline: Pipeline) -> Image:
|
40 |
empty_cache()
|
41 |
generator = Generator("cuda").manual_seed(request.seed)
|
42 |
-
image=pipeline(request.prompt,generator=generator, guidance_scale=0.0, num_inference_steps=
|
43 |
return(image)
|
|
|
39 |
def infer(request: TextToImageRequest, pipeline: Pipeline) -> Image:
|
40 |
empty_cache()
|
41 |
generator = Generator("cuda").manual_seed(request.seed)
|
42 |
+
image=pipeline(request.prompt,generator=generator, guidance_scale=0.0, num_inference_steps=4, max_sequence_length=256, height=request.height, width=request.width, output_type="pil").images[0]
|
43 |
return(image)
|