Spaces:

SaitoHoujou
/

Image-to-Sketch_Translation_System_with_Finetuned-ControlNet_for_Diffusion_Model

Sleeping

SaitoHoujou commited on 29 days ago

Commit

8aa62c7

verified ·

1 Parent(s): b6ada02

Update StableDiffusionControlNetPipeline to StableDiffusion3ControlNetPipeline

Files changed (1) hide show

streamlit_app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import streamlit as st
 from PIL import Image
 import torch
-from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, UniPCMultistepScheduler
 from controlnet_aux import HEDdetector
 app_controlnet_path = "SaitoHoujou/Finetuned-ControlNet_for_Diffusion-Model"
@@ -55,7 +55,7 @@ def load_pipeline():
         torch_dtype=app_dtype
     )
-    pipe = StableDiffusionControlNetPipeline.from_pretrained(
         app_base_model,
         controlnet=controlnet,
         torch_dtype=app_dtype,

 import streamlit as st
 from PIL import Image
 import torch
+from diffusers import StableDiffusion3ControlNetPipeline, ControlNetModel, UniPCMultistepScheduler
 from controlnet_aux import HEDdetector
 app_controlnet_path = "SaitoHoujou/Finetuned-ControlNet_for_Diffusion-Model"
         torch_dtype=app_dtype
     )
+    pipe = StableDiffusion3ControlNetPipeline.from_pretrained(
         app_base_model,
         controlnet=controlnet,
         torch_dtype=app_dtype,