diff --git a/scripts/sample.sh b/scripts/sample.sh index fe5093daf..6c82d319e 100644 --- a/scripts/sample.sh +++ b/scripts/sample.sh @@ -11,5 +11,5 @@ accelerate launch \ --image-size 256 \ --num-sampling-steps 250 \ --attention-mode flash \ - -mixed_precision bf16 + --mixed-precision bf16