Skip to content

Commit 922d56a

Browse files
Correct type from int to str in conversion script sd
1 parent ae85474 commit 922d56a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scripts/convert_original_stable_diffusion_to_diffusers.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -694,7 +694,7 @@ def convert_open_clip_checkpoint(checkpoint):
694694
parser.add_argument(
695695
"--prediction_type",
696696
default=None,
697-
type=int,
697+
type=str,
698698
help=(
699699
"The prediction type that the model was trained on. Use 'epsilon' for Stable Diffusion v1.X and Stable"
700700
" Siffusion v2 Base. Use 'v-prediction' for Stable Diffusion v2."

0 commit comments

Comments
 (0)