initial SP4D commit

This commit is contained in:
Chun-Han Yao
2025-11-03 21:39:45 +00:00
parent fd9d14e02f
commit 0aee97d395
2 changed files with 13 additions and 1 deletions

View File

@@ -54,7 +54,7 @@ def sample(
model_path: Optional[str] = "checkpoints/sp4d.safetensors",
output_folder: Optional[str] = "outputs",
num_steps: Optional[int] = 50,
img_size: int = 576, # image resolution
img_size: int = 512, # image resolution
n_frames: int = 4, # number of input and output video frames
seed: int = 23,
encoding_t: int = 8, # Number of frames encoded at a time! This eats most VRAM. Reduce if necessary.