diff --git a/tutorials/generative/3d_ddpm/3d_ddpm_tutorial.ipynb b/tutorials/generative/3d_ddpm/3d_ddpm_tutorial.ipynb
index 0c3ff8a5..188d78d5 100644
--- a/tutorials/generative/3d_ddpm/3d_ddpm_tutorial.ipynb
+++ b/tutorials/generative/3d_ddpm/3d_ddpm_tutorial.ipynb
@@ -354,7 +354,7 @@
     "    out_channels=1,\n",
     "    num_channels=[256, 256, 512],\n",
     "    attention_levels=[False, False, True],\n",
-    "    num_head_channels=[256, 256, 512],\n",
+    "    num_head_channels=[0, 0, 512],\n",
     "    num_res_blocks=2,\n",
     ")\n",
     "model.to(device)\n",
diff --git a/tutorials/generative/3d_ddpm/3d_ddpm_tutorial.py b/tutorials/generative/3d_ddpm/3d_ddpm_tutorial.py
index 612b84dd..41c4960f 100644
--- a/tutorials/generative/3d_ddpm/3d_ddpm_tutorial.py
+++ b/tutorials/generative/3d_ddpm/3d_ddpm_tutorial.py
@@ -149,7 +149,7 @@
     out_channels=1,
     num_channels=[256, 256, 512],
     attention_levels=[False, False, True],
-    num_head_channels=[256, 256, 512],
+    num_head_channels=[0, 0, 512],
     num_res_blocks=2,
 )
 model.to(device)