Update gen_demo.py

gen_demo.py

@@ -119,7 +119,7 @@ num_slots_to_gen = 32
 # Function to generate image from class
 def generate_from_class(class_id, cfg_scale):
     with torch.no_grad():
-        dtype = torch.bfloat16
+        dtype = torch.float
         num_slots_to_gen = 32
         with torch.autocast(device, dtype=dtype):
             slots_gen = generate(