huggingface · ayushtues · May 15, 2023 · May 17, 2023 · May 18, 2023 · May 22, 2023
diff --git a/examples/consistency_models/requirements.txt b/examples/consistency_models/requirements.txt
@@ -0,0 +1,3 @@
+accelerate>=0.16.0
+torchvision
+datasets
diff --git a/examples/consistency_models/script.sh b/examples/consistency_models/script.sh
@@ -0,0 +1,3 @@
+#!/bin/bash
+
+accelerate launch train_consistency_distillation.py --dataset_name="cifar10" --resolution=32 --center_crop --random_flip  --output_dir="cifar10-32" --train_batch_size=16 --num_epochs=100 --gradient_accumulation_steps=1 --learning_rate=1e-4 --lr_warmup_steps=500 --mixed_precision=no --push_to_hub
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		#!/bin/bash
Copy link Member sayakpaul Jul 10, 2023 Choose a reason for hiding this comment The reason will be displayed to describe this comment to others. Learn more. We don't include shell-scripts in the examples.

		accelerate launch train_consistency_distillation.py --dataset_name="cifar10" --resolution=32 --center_crop --random_flip --output_dir="cifar10-32" --train_batch_size=16 --num_epochs=100 --gradient_accumulation_steps=1 --learning_rate=1e-4 --lr_warmup_steps=500 --mixed_precision=no --push_to_hub