remove the comment, remove diffusers in requiremnets.txt, add validation_image ote

lavinal712 · lavinal712 · commit 80d39797b9b9 · 2025-01-22T01:47:24.000Z
diff --git a/examples/research_projects/autoencoderkl/README.md b/examples/research_projects/autoencoderkl/README.md
@@ -27,6 +27,8 @@ accelerate config
 
 ## Training on CIFAR10
 
+Please replace the validation image with your own image.
+
 ```bash
 accelerate launch train_autoencoderkl.py \
     --pretrained_model_name_or_path stabilityai/sd-vae-ft-mse \
diff --git a/examples/research_projects/autoencoderkl/requirements.txt b/examples/research_projects/autoencoderkl/requirements.txt
@@ -1,7 +1,6 @@
 accelerate>=0.16.0
 bitsandbytes
 datasets
-diffusers
 huggingface_hub
 lpips
 numpy
diff --git a/examples/research_projects/autoencoderkl/train_autoencoderkl.py b/examples/research_projects/autoencoderkl/train_autoencoderkl.py
@@ -58,7 +58,7 @@
     import wandb
 
 # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
-# check_min_version("0.33.0.dev0")
+check_min_version("0.33.0.dev0")
 
 logger = get_logger(__name__)
 
