KawaiiApp
/

anythinv3-vae-handler

StableDiffusionPipeline

stable-diffusion

stable-diffusion-diffusers

Model card Files Files and versions

KawaiiApp commited on Jun 25, 2023

Commit

1997091

·

1 Parent(s): 6ad32f2

removed input

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -40,9 +40,9 @@ class EndpointHandler():
         # run inference pipeline
         with autocast(device.type):
             if negative_prompt is None:
-                image = self.pipe(inputs,prompt = postive_prompt ,height = height ,width = width ,guidance_scale=float(guidance_scale))["sample"][0]
             else:
-                image = self.pipe(inputs,prompt = postive_prompt ,negative_prompt = negative_prompt,height = height ,width = width ,guidance_scale=float(guidance_scale))["sample"][0]
         # encode image as base 64
         buffered = BytesIO()

         # run inference pipeline
         with autocast(device.type):
             if negative_prompt is None:
+                image = self.pipe(prompt = postive_prompt ,height = height ,width = width ,guidance_scale=float(guidance_scale))["sample"][0]
             else:
+                image = self.pipe(prompt = postive_prompt ,negative_prompt = negative_prompt,height = height ,width = width ,guidance_scale=float(guidance_scale))["sample"][0]
         # encode image as base 64
         buffered = BytesIO()