n_sample memory collect

gradio_utils/motionctrl_cmcm_gradio.py

@@ -184,10 +184,11 @@ def motionctrl_sample(
                 model.en_and_decode_n_samples_a_time = decoding_t
                 samples_x = model.decode_first_stage(samples_z)
                 samples = torch.clamp((samples_x + 1.0) / 2.0, min=0.0, max=1.0) # [1*t, c, h, w]
+                samples = samples.data.cpu()
                 results.append(samples)
 
             samples = torch.stack(results, dim=0) # [sample_num, t, c, h, w]
-            samples = samples.data.cpu()
+            # samples = samples.data.cpu()
 
             video_path = tempfile.NamedTemporaryFile(suffix='.mp4').name
             save_results(samples, video_path, fps=save_fps)