From 6a4bf3e687926c7af8b601b54a6b8f79be51c632 Mon Sep 17 00:00:00 2001
From: Ktosh100001 <107479736+Ktosh100001@users.noreply.github.com>
Date: Thu, 26 Jun 2025 18:26:55 +0530
Subject: [PATCH] Update script_util.py

---
 guided_diffusion/script_util.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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():