Update scripts/evaluation/funcs.py

scripts/evaluation/funcs.py

@@ -79,9 +79,9 @@ 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, v.dtype)
+        print(samples.device, samples.dtype)
         ## reconstruct from latent to pixel space
         batch_images = model.decode_first_stage(samples, **additional_decode_kwargs)