silencer107 commited on
Commit
ecfc08b
·
verified ·
1 Parent(s): 83993a2

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. 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=3, max_sequence_length=256, height=request.height, width=request.width, output_type="pil").images[0]
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)