Function
_get_add_time_ids
(
fps,
motion_bucket_id,
noise_aug_strength,
dtype,
batch_size,
unet=None,
device=None
)
Source from the content-addressed store, hash-verified
| 1233 | |
| 1234 | |
| 1235 | def _get_add_time_ids( |
| 1236 | fps, |
| 1237 | motion_bucket_id, |
| 1238 | noise_aug_strength, |
| 1239 | dtype, |
| 1240 | batch_size, |
| 1241 | unet=None, |
| 1242 | device=None |
| 1243 | ): |
| 1244 | add_time_ids = [fps, motion_bucket_id, noise_aug_strength] |
| 1245 | |
| 1246 | |
| 1247 | add_time_ids = torch.tensor([add_time_ids], dtype=dtype, device=device) |
| 1248 | add_time_ids = add_time_ids.repeat(batch_size, 1) |
| 1249 | return add_time_ids |
| 1250 | |
| 1251 | # Potentially load in the weights and states from a previous save |
| 1252 | if args.resume_from_checkpoint: |
Tested by
no test coverage detected