KawaiiApp
/

anythinv3-vae-handler

StableDiffusionPipeline

stable-diffusion

stable-diffusion-diffusers

Model card Files Files and versions

KawaiiApp commited on Jun 25, 2023

Commit

492876f

·

1 Parent(s): a632781

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -42,11 +42,11 @@ class EndpointHandler():
         # Run inference pipeline
         with autocast(device.type):
             if negative_prompt is None:
-                print(prompt=inputs, height=height, width=width, guidance_scale=float(guidance_scale))
                 image = self.pipe(prompt=inputs, height=height, width=width, guidance_scale=float(guidance_scale))
                 image = image.images[0]
             else:
-                print(prompt=inputs, height=height, width=width, guidance_scale=float(guidance_scale))
                 image = self.pipe(prompt=inputs, negative_prompt=negative_prompt, height=height, width=width, guidance_scale=float(guidance_scale))
                 image = image.images[0]

         # Run inference pipeline
         with autocast(device.type):
             if negative_prompt is None:
+                print(str(inputs), str(height), str(width), str(guidance_scale_)
                 image = self.pipe(prompt=inputs, height=height, width=width, guidance_scale=float(guidance_scale))
                 image = image.images[0]
             else:
+                print(str(inputs), str(height), str(negative_prompt), str(width), str(guidance_scale))
                 image = self.pipe(prompt=inputs, negative_prompt=negative_prompt, height=height, width=width, guidance_scale=float(guidance_scale))
                 image = image.images[0]