From e82a6e0f3da81a73e1ee937d0b310be18982fca6 Mon Sep 17 00:00:00 2001
From: Yuan-Man <68322456+Yuan-ManX@users.noreply.github.com>
Date: Mon, 10 Mar 2025 14:41:28 +0800
Subject: [PATCH] Create style_crafter_sdxl.yaml

---
 .../config/infer/style_crafter_sdxl.yaml      | 31 +++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100644 StyleStudio/stylecrafter_sdxl/config/infer/style_crafter_sdxl.yaml

diff --git a/StyleStudio/stylecrafter_sdxl/config/infer/style_crafter_sdxl.yaml b/StyleStudio/stylecrafter_sdxl/config/infer/style_crafter_sdxl.yaml
new file mode 100644
index 0000000..e870bca
--- /dev/null
+++ b/StyleStudio/stylecrafter_sdxl/config/infer/style_crafter_sdxl.yaml
@@ -0,0 +1,31 @@
+model:
+  pretrained_model_name_or_path: path/to/sdxl
+  image_encoder_path: path/to/stylecrafter-sdxl/image_encoder
+  model: 
+    target: models.stylecrafter.StyleCrafter
+    params:
+      scale: 1.0
+      use_norm: true
+      text_dim: 2048
+      style_dim: 2048
+      image_context_config:
+        target: models.encoder.QueryTransformer
+        params:
+          embedding_dim: 1280
+          output_dim: 2048
+          num_heads: 20
+          num_queries: 16
+          n_layers: 4
+      scale_predictor_config:
+        target: models.encoder.QueryTransformer
+        params:
+          embedding_dim: 512
+          output_dim: 512
+          num_heads: 8
+          num_queries: 70
+          n_layers: 2
+
+pretrained: path/to/stylecrafter-sdxl/stylecrafter/stylecrafter_sdxl.ckpt
+height: 1024
+width: 1024
+steps: 50