Commit
·
1e9a1ec
1
Parent(s):
817cd1e
remove printing of latent shape
Browse files
cosmos_transfer1/diffusion/inference/world_generation_pipeline.py
CHANGED
@@ -553,7 +553,6 @@ class DiffusionControl2WorldGenerationPipeline(BaseWorldGenerationPipeline):
|
|
553 |
end_frame = num_new_generated_frames * (i_clip + 1) + self.num_input_frames
|
554 |
|
555 |
# Prepare x_sigma_max
|
556 |
-
print("==============================================================")
|
557 |
if input_video is not None:
|
558 |
if is_upscale_case:
|
559 |
x_sigma_max = []
|
@@ -573,8 +572,6 @@ class DiffusionControl2WorldGenerationPipeline(BaseWorldGenerationPipeline):
|
|
573 |
|
574 |
else:
|
575 |
x_sigma_max = None
|
576 |
-
print("final ->", x_sigma_max.shape)
|
577 |
-
print("==============================================================")
|
578 |
|
579 |
data_batch_i[hint_key] = control_input[:, :, start_frame:end_frame].cuda()
|
580 |
latent_hint = []
|
|
|
553 |
end_frame = num_new_generated_frames * (i_clip + 1) + self.num_input_frames
|
554 |
|
555 |
# Prepare x_sigma_max
|
|
|
556 |
if input_video is not None:
|
557 |
if is_upscale_case:
|
558 |
x_sigma_max = []
|
|
|
572 |
|
573 |
else:
|
574 |
x_sigma_max = None
|
|
|
|
|
575 |
|
576 |
data_batch_i[hint_key] = control_input[:, :, start_frame:end_frame].cuda()
|
577 |
latent_hint = []
|
cosmos_transfer1/diffusion/model/model_t2w.py
CHANGED
@@ -177,9 +177,6 @@ class DiffusionT2WModel(torch.nn.Module):
|
|
177 |
noise prediction (eps_pred) and optional confidence (logvar).
|
178 |
"""
|
179 |
|
180 |
-
print("<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<")
|
181 |
-
print(xt.shape)
|
182 |
-
print("<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<")
|
183 |
xt = xt.to(**self.tensor_kwargs)
|
184 |
sigma = sigma.to(**self.tensor_kwargs)
|
185 |
# get precondition for the network
|
|
|
177 |
noise prediction (eps_pred) and optional confidence (logvar).
|
178 |
"""
|
179 |
|
|
|
|
|
|
|
180 |
xt = xt.to(**self.tensor_kwargs)
|
181 |
sigma = sigma.to(**self.tensor_kwargs)
|
182 |
# get precondition for the network
|