OmPrakashSingh1704 commited on
Commit
0c13c1c
·
1 Parent(s): 47835d7
options/Video_model/Model.py CHANGED
@@ -10,7 +10,7 @@ pipeline = StableVideoDiffusionPipeline.from_pretrained(
10
  ).to(device)
11
  # pipeline.enable_model_cpu_offload()
12
  def Video(image):
13
- # image = Image.open(image)
14
  image = image.resize((1024, 576))
15
 
16
  generator = torch.Generator(device=device).manual_seed(42)
 
10
  ).to(device)
11
  # pipeline.enable_model_cpu_offload()
12
  def Video(image):
13
+ image = Image.fromarray(image)
14
  image = image.resize((1024, 576))
15
 
16
  generator = torch.Generator(device=device).manual_seed(42)
options/Video_model/__pycache__/Model.cpython-310.pyc CHANGED
Binary files a/options/Video_model/__pycache__/Model.cpython-310.pyc and b/options/Video_model/__pycache__/Model.cpython-310.pyc differ