tchoudha21 commited on
Commit
b7bc788
·
verified ·
1 Parent(s): 5604d40

added image input path

Browse files
Files changed (1) hide show
  1. 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", None)
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):