Commit 922d56a1 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

Correct type from int to str in conversion script sd

parent ae854746
......@@ -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."
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment