-
Notifications
You must be signed in to change notification settings - Fork 3k
Open
Description
Running the following command:
python scripts/sampling/simple_video_sample.py --version sv3d_u --verbose --decoding_t 1
The program starts, but then it seems to just die suddenly with no warnings near the very end of inference:
...
VideoTransformerBlock is using checkpointing
Initialized embedder #0: FrozenOpenCLIPImagePredictionEmbedder with 683800065 params. Trainable: False
Initialized embedder #1: VideoPredictionEmbedderWithEncoder with 83653863 params. Trainable: False
Initialized embedder #2: ConcatTimestepEmbedderND with 0 params. Trainable: False
Restored from checkpoints/sv3d_u.safetensors with 0 missing and 0 unexpected keys
############################## Sampling setting ##############################
Sampler: EulerEDMSampler
Discretization: EDMDiscretization
Guider: TrianglePredictionGuider
Sampling with EulerEDMSampler for 51 steps: 0%| | 0/51 [00:00<?, ?it/s]/home/captdishwasher/horenbergerb/sv3d/generative-models/.pt2/lib/python3.10/site-packages/torch/utils/checkpoint.py:31: UserWarning: None of the inputs have requires_grad=True. Gradients will be None
warnings.warn("None of the inputs have requires_grad=True. Gradients will be None")
Sampling with EulerEDMSampler for 51 steps: 98%|████████████████████████████████████████████████████████████████████████████████████████████████████████▉ | 50/51 [01:31<00:01, 1.83s/it]
(.pt2) (base) captdishwasher@captainofthedishwasher-MS-7D43:~/horenbergerb/sv3d/generative-models$
Sometimes the output folder is empty, and sometimes it contains a single jpg and a ~20mb mp4 which is corrupted and won't play.
Is there something I'm missing? Is this a VRAM problem? I'm running 24GB of VRAM on an NVIDIA 3090, Ubuntu 22.04.
Thanks!
elihalpern, Ashoka74, eugenefrunze, FlonneEx, ayan4m1 and 1 more
Metadata
Metadata
Assignees
Labels
No labels