xizaoqu
commited on
Commit
·
be19e81
1
Parent(s):
b799cfa
update
Browse files
app.py
CHANGED
|
@@ -200,6 +200,11 @@ def run_interactive(first_frame, action, first_pose, curr_frame, device, self_fr
|
|
| 200 |
self_memory_c2w=self_memory_c2w,
|
| 201 |
self_frame_idx=self_frame_idx)
|
| 202 |
print("in run_interactive")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 203 |
return new_frame, self_frames, self_poses, self_memory_c2w, self_frame_idx
|
| 204 |
|
| 205 |
def set_denoising_steps(denoising_steps, sampling_timesteps_state):
|
|
|
|
| 200 |
self_memory_c2w=self_memory_c2w,
|
| 201 |
self_frame_idx=self_frame_idx)
|
| 202 |
print("in run_interactive")
|
| 203 |
+
print(new_frame.device)
|
| 204 |
+
print(self_frames.device)
|
| 205 |
+
print(self_poses.device)
|
| 206 |
+
print(self_memory_c2w.device)
|
| 207 |
+
print(self_frame_idx.device)
|
| 208 |
return new_frame, self_frames, self_poses, self_memory_c2w, self_frame_idx
|
| 209 |
|
| 210 |
def set_denoising_steps(denoising_steps, sampling_timesteps_state):
|