From 51025b56c90d4a5d90fdb36831e16dd44ec0516d Mon Sep 17 00:00:00 2001 From: Victor Hall Date: Wed, 8 Feb 2023 20:39:36 -0500 Subject: [PATCH] fix sd21 conversion in colab --- Train_Colab.ipynb | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/Train_Colab.ipynb b/Train_Colab.ipynb index 553cba1..86e2f5d 100644 --- a/Train_Colab.ipynb +++ b/Train_Colab.ipynb @@ -192,6 +192,7 @@ "\n", "img_size = 512\n", "upscale_attention = False\n", + "prediction_type = \"epsilon\"\n", "if model_type == \"SD1x\":\n", " inference_yaml = \"v1-inference.yaml\"\n", "elif model_type == \"SD2_512_base\":\n", @@ -199,6 +200,7 @@ " inference_yaml = \"v2-inference.yaml\"\n", "elif model_type == \"SD21\":\n", " upscale_attention = True\n", + " prediction_type = \"v_prediction\"\n", " inference_yaml = \"v2-inference-v.yaml\"\n", " img_size = 768\n", "\n", @@ -209,7 +211,7 @@ "--original_config_file {inference_yaml} \\\n", "--image_size {img_size} \\\n", "--checkpoint_path {base_path} \\\n", - "--prediction_type epsilon \\\n", + "--prediction_type {prediction_type} \\\n", "--upcast_attn False \\\n", "--dump_path {save_name}" ]