added image input path
Browse files- video2world_hf.py +1 -1
video2world_hf.py
CHANGED
|
@@ -41,7 +41,7 @@ class DiffusionVideo2WorldConfig(PretrainedConfig):
|
|
| 41 |
|
| 42 |
self.diffusion_transformer_dir = kwargs.get("diffusion_transformer_dir", "Cosmos-1.0-Diffusion-7B-Video2World")
|
| 43 |
self.prompt_upsampler_dir = kwargs.get("prompt_upsampler_dir", "Pixtral-12B")
|
| 44 |
-
self.input_image_or_video_path = kwargs.get("input_image_or_video_path",
|
| 45 |
self.num_input_frames = kwargs.get("num_input_frames", 1)
|
| 46 |
|
| 47 |
class DiffusionVideo2World(PreTrainedModel):
|
|
|
|
| 41 |
|
| 42 |
self.diffusion_transformer_dir = kwargs.get("diffusion_transformer_dir", "Cosmos-1.0-Diffusion-7B-Video2World")
|
| 43 |
self.prompt_upsampler_dir = kwargs.get("prompt_upsampler_dir", "Pixtral-12B")
|
| 44 |
+
self.input_image_or_video_path = kwargs.get("input_image_or_video_path", "inputs/input.jpg")
|
| 45 |
self.num_input_frames = kwargs.get("num_input_frames", 1)
|
| 46 |
|
| 47 |
class DiffusionVideo2World(PreTrainedModel):
|