Update scripts/evaluation/funcs.py

scripts/evaluation/funcs.py

@@ -79,8 +79,8 @@ def batch_ddim_sampling(model, cond, noise_shape, n_samples=1, ddim_steps=50, dd
         torch.cuda.empty_cache()
         model.cpu()
         model.first_stage_model.cuda()
-        for k, v in model.named_parameters():
-            print(k, v.device)
+        # for k, v in model.named_parameters():
+            # print(k, v.device)
         
         ## reconstruct from latent to pixel space
         batch_images = model.decode_first_stage(samples, **additional_decode_kwargs)