Charles-Elena
/

ControlNet-endpoint-test

stable-diffusion

stable-diffusion-diffusers

endpoints-template

Inference Endpoints

Model card Files Files and versions Community

charleselena commited on Jan 21

Commit

3eb4425

•

1 Parent(s): 7090d3d

test lib imports

Files changed (1) hide show

handler.py +3 -3

handler.py CHANGED Viewed

@@ -2,11 +2,11 @@ from typing import Dict, List, Any
 import base64
 from PIL import Image
 from io import BytesIO
-#from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
-from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, StableDiffusionSafetyChecker
 # import Safety Checker
 # from transformers import AutoProcessor, SafetyChecker
-#from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
 import torch

 import base64
 from PIL import Image
 from io import BytesIO
+from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
+#from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, StableDiffusionSafetyChecker
 # import Safety Checker
 # from transformers import AutoProcessor, SafetyChecker
+from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
 import torch