KawaiiApp
/

anythinv3-vae-handler

StableDiffusionPipeline

stable-diffusion

stable-diffusion-diffusers

Inference Endpoints

Model card Files Files and versions Community

KawaiiApp commited on Jun 25, 2023

Commit

a9108ee

·

1 Parent(s): 492876f

refactor

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -42,7 +42,7 @@ class EndpointHandler():
         # 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:

         # 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: