diff --git a/pages/1_Image_Creator.py b/pages/1_Image_Creator.py index 2b50072..d5a26de 100644 --- a/pages/1_Image_Creator.py +++ b/pages/1_Image_Creator.py @@ -5,7 +5,7 @@ import streamlit as st st.set_page_config(layout="wide") from streamlit_drawable_canvas import st_canvas from diffsynth.models import ModelManager -from diffsynth.pipelines import SDImagePipeline, SDXLImagePipeline, HunyuanDiTImagePipeline +from diffsynth.pipelines import SDImagePipeline, SDXLImagePipeline, SD3ImagePipeline, HunyuanDiTImagePipeline from diffsynth.data.video import crop_and_resize @@ -20,6 +20,11 @@ config = { "pipeline_class": SDXLImagePipeline, "fixed_parameters": {} }, + "Stable Diffusion 3": { + "model_folder": "models/stable_diffusion_3", + "pipeline_class": SD3ImagePipeline, + "fixed_parameters": {} + }, "Stable Diffusion XL Turbo": { "model_folder": "models/stable_diffusion_xl_turbo", "pipeline_class": SDXLImagePipeline,