diff --git a/guided_diffusion/script_util.py b/guided_diffusion/script_util.py
index 2bfdad9fc..d36137766 100644
--- a/guided_diffusion/script_util.py
+++ b/guided_diffusion/script_util.py
@@ -5,7 +5,7 @@
 from .respace import SpacedDiffusion, space_timesteps
 from .unet import SuperResModel, UNetModel, EncoderUNetModel
 
-NUM_CLASSES = 1000
+NUM_CLASSES = 10
 
 
 def diffusion_defaults():