minor cghange

handler.py

@@ -29,7 +29,7 @@ class EndpointHandler():
 # import torch
 
  device = "cuda"
- num_images_per_prompt = 2
+ num_images_per_prompt = 1
 
  prior = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=torch.bfloat16).to(device)
  decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.float16).to(device)
@@ -39,10 +39,10 @@ class EndpointHandler():
 
  prior_output = prior(
  prompt=prompt,
- height=1024,
- width=1024,
+ height=512,
+ width=512,
  negative_prompt=negative_prompt,
- guidance_scale=4.0,
+ guidance_scale=7.0,
  num_images_per_prompt=num_images_per_prompt,
  num_inference_steps=20
  )
@@ -50,11 +50,11 @@ class EndpointHandler():
  image_embeddings=prior_output.image_embeddings.half(),
  prompt=prompt,
  negative_prompt=negative_prompt,
- guidance_scale=0.0,
+ guidance_scale=7.0,
  output_type="pil",
  num_inference_steps=10
  ).images
- return decoder_output
+ return decoder_output[0]