Merge pull request #158 from lin-tianyu/master

WuJunde · web-flow · commit 236fe40e0dd8 · 2024-03-10T14:04:24.000Z
solve DPM-Solver OOM issue
diff --git a/guided_diffusion/gaussian_diffusion.py b/guided_diffusion/gaussian_diffusion.py
@@ -549,6 +549,7 @@ def p_sample_loop_known(
                 skip_type="time_uniform",
                 method="multistep",
             )
+            sample = sample.detach()    ### MODIFIED: for DPM-Solver OOM issue
             sample[:,-1,:,:] = norm(sample[:,-1,:,:])
             final["sample"] = sample
             final["cal"] = cal

Original file line number	Diff line number	Diff line change
`@@ -549,6 +549,7 @@ def p_sample_loop_known(`
`549`	`549`	`skip_type="time_uniform",`
`550`	`550`	`method="multistep",`
`551`	`551`	`)`
	`552`	`+ sample = sample.detach() ### MODIFIED: for DPM-Solver OOM issue`
`552`	`553`	`sample[:,-1,:,:] = norm(sample[:,-1,:,:])`
`553`	`554`	`final["sample"] = sample`
`554`	`555`	`final["cal"] = cal`