Correct type from int to str in conversion script sd

922d56a1 · Patrick von Platen · ae854746 · 922d56a1
Commit 922d56a1 authored Dec 05, 2022 by Patrick von Platen
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

scripts/convert_original_stable_diffusion_to_diffusers.py scripts/convert_original_stable_diffusion_to_diffusers.py +1 -1

No files found.
--- a/scripts/convert_original_stable_diffusion_to_diffusers.py
+++ b/scripts/convert_original_stable_diffusion_to_diffusers.py
@@ -694,7 +694,7 @@ if __name__ == "__main__":
    parser.add_argument(
        "--prediction_type",
        default=None,
-        type=int,
+        type=str,
        help=(
            "The prediction type that the model was trained on. Use 'epsilon' for Stable Diffusion v1.X and Stable"
            " Siffusion v2 Base. Use 'v-prediction' for Stable Diffusion v2."