From 6dc019d71ab92813b23bb128d567a28a93b6fbae Mon Sep 17 00:00:00 2001 From: Valay Dave Date: Mon, 4 Dec 2023 23:06:56 -0800 Subject: [PATCH] SD XL Integration --- config.py | 2 +- diffusion.py | 2 +- style_config.yaml | 4 ++-- style_video.yaml | 4 ++-- text_config.yaml | 4 ++-- video_config.yaml | 4 ++-- 6 files changed, 10 insertions(+), 10 deletions(-) diff --git a/config.py b/config.py index e418e8d..8a5a4c7 100644 --- a/config.py +++ b/config.py @@ -7,7 +7,7 @@ VIDEO_MODEL_NAME = "stable-video-diffusion-img2vid" MODELS_BASE_S3_PATH = "models/diffusion-models/" -IMAGE_MODEL_NAME = "stable-diffusion-xl-base-1.0.1" +IMAGE_MODEL_NAME = "sdxl-turbo" MODEL_PATH = "./models" DEFAULT_STYLES = [ "van gogh", diff --git a/diffusion.py b/diffusion.py index 13ed39e..1092562 100644 --- a/diffusion.py +++ b/diffusion.py @@ -4,7 +4,7 @@ import os from diffusers import AutoPipelineForText2Image -SD_XL_BASE = "stable-diffusion-xl-base-1.0" +SD_XL_BASE = "sdxl-turbo" SUPPORTED_PIPELINES = ["StableDiffusionXLPipeline", "StableDiffusionPipeline"] IMAGE_MODEL_NAME = SD_XL_BASE diff --git a/style_config.yaml b/style_config.yaml index 7c27a6b..241fd98 100644 --- a/style_config.yaml +++ b/style_config.yaml @@ -3,8 +3,8 @@ model_config: pretrained_model_path: ./models force_upload: false s3_prefix: models/diffusion-models/ - model_version: stable-diffusion-xl-base-1.0.1 - model_name: stable-diffusion-xl-base-1.0.1 + model_version: sdxl-turbo + model_name: sdxl-turbo inference_config: batch_size: 1 height: 512 diff --git a/style_video.yaml b/style_video.yaml index ee4b39f..0fdca2c 100644 --- a/style_video.yaml +++ b/style_video.yaml @@ -26,8 +26,8 @@ image: pretrained_model_path: ./models force_upload: false s3_prefix: models/diffusion-models/ - model_version: stable-diffusion-xl-base-1.0.1 - model_name: stable-diffusion-xl-base-1.0.1 + model_version: sdxl-turbo + model_name: sdxl-turbo inference_config: batch_size: 2 height: 576 diff --git a/text_config.yaml b/text_config.yaml index 6b5ff4c..b6e3709 100644 --- a/text_config.yaml +++ b/text_config.yaml @@ -3,8 +3,8 @@ model_config: pretrained_model_path: ./models force_upload: false s3_prefix: models/diffusion-models/ - model_version: stable-diffusion-xl-base-1.0.1 - model_name: stable-diffusion-xl-base-1.0.1 + model_version: sdxl-turbo + model_name: sdxl-turbo inference_config: batch_size: 1 height: 512 diff --git a/video_config.yaml b/video_config.yaml index ecaf53b..1d6b1b5 100644 --- a/video_config.yaml +++ b/video_config.yaml @@ -26,8 +26,8 @@ image: pretrained_model_path: ./models force_upload: false s3_prefix: models/diffusion-models/ - model_version: stable-diffusion-xl-base-1.0.1 - model_name: stable-diffusion-xl-base-1.0.1 + model_version: sdxl-turbo + model_name: sdxl-turbo inference_config: batch_size: 2 height: 576