Update configs/hair_transfer.yaml
Browse fileschanged the stable diffusion model
configs/hair_transfer.yaml
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
pretrained_model_path: "stabilityai/stable-diffusion-2-
|
| 2 |
|
| 3 |
pretrained_folder: "LogicGoInfotechSpaces/new_weights/stage2"
|
| 4 |
encoder_path: "pytorch_model.bin"
|
|
|
|
| 1 |
+
pretrained_model_path: "stabilityai/stable-diffusion-2-base" # Use base instead of 2-1-base for compatibility
|
| 2 |
|
| 3 |
pretrained_folder: "LogicGoInfotechSpaces/new_weights/stage2"
|
| 4 |
encoder_path: "pytorch_model.bin"
|