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]