Hugging Face's logo

Spaces:
SherryX
/
STAR
Running on Zero

App Files Community
6
Solitude-liu commited on
Commit
812ca1b
·
1 Parent(s): d2b5b0e

fix file error

Browse files
Files changed (18) hide show
  1. __pycache__/inference_utils.cpython-39.pyc +0 -0
  2. pretrained_weight/__put_model_here__ +0 -0
  3. video_super_resolution/__pycache__/color_fix.cpython-39.pyc +0 -0
  4. video_super_resolution/scripts/__pycache__/inference_sr.cpython-39.pyc +0 -0
  5. video_to_video/__pycache__/__init__.cpython-39.pyc +0 -0
  6. video_to_video/__pycache__/video_to_video_model.cpython-39.pyc +0 -0
  7. video_to_video/diffusion/__pycache__/__init__.cpython-39.pyc +0 -0
  8. video_to_video/diffusion/__pycache__/diffusion_sdedit.cpython-39.pyc +0 -0
  9. video_to_video/diffusion/__pycache__/schedules_sdedit.cpython-39.pyc +0 -0
  10. video_to_video/diffusion/__pycache__/solvers_sdedit.cpython-39.pyc +0 -0
  11. video_to_video/modules/__pycache__/__init__.cpython-39.pyc +0 -0
  12. video_to_video/modules/__pycache__/embedder.cpython-39.pyc +0 -0
  13. video_to_video/modules/__pycache__/unet_v2v.cpython-39.pyc +0 -0
  14. video_to_video/utils/__pycache__/__init__.cpython-39.pyc +0 -0
  15. video_to_video/utils/__pycache__/config.cpython-39.pyc +0 -0
  16. video_to_video/utils/__pycache__/logger.cpython-39.pyc +0 -0
  17. video_to_video/utils/__pycache__/seed.cpython-39.pyc +0 -0
  18. video_to_video/video_to_video_model.py +5 -5
__pycache__/inference_utils.cpython-39.pyc CHANGED
Binary files a/__pycache__/inference_utils.cpython-39.pyc and b/__pycache__/inference_utils.cpython-39.pyc differ
 
pretrained_weight/__put_model_here__ ADDED
File without changes
video_super_resolution/__pycache__/color_fix.cpython-39.pyc CHANGED
Binary files a/video_super_resolution/__pycache__/color_fix.cpython-39.pyc and b/video_super_resolution/__pycache__/color_fix.cpython-39.pyc differ
 
video_super_resolution/scripts/__pycache__/inference_sr.cpython-39.pyc CHANGED
Binary files a/video_super_resolution/scripts/__pycache__/inference_sr.cpython-39.pyc and b/video_super_resolution/scripts/__pycache__/inference_sr.cpython-39.pyc differ
 
video_to_video/__pycache__/__init__.cpython-39.pyc CHANGED
Binary files a/video_to_video/__pycache__/__init__.cpython-39.pyc and b/video_to_video/__pycache__/__init__.cpython-39.pyc differ
 
video_to_video/__pycache__/video_to_video_model.cpython-39.pyc CHANGED
Binary files a/video_to_video/__pycache__/video_to_video_model.cpython-39.pyc and b/video_to_video/__pycache__/video_to_video_model.cpython-39.pyc differ
 
video_to_video/diffusion/__pycache__/__init__.cpython-39.pyc CHANGED
Binary files a/video_to_video/diffusion/__pycache__/__init__.cpython-39.pyc and b/video_to_video/diffusion/__pycache__/__init__.cpython-39.pyc differ
 
video_to_video/diffusion/__pycache__/diffusion_sdedit.cpython-39.pyc CHANGED
Binary files a/video_to_video/diffusion/__pycache__/diffusion_sdedit.cpython-39.pyc and b/video_to_video/diffusion/__pycache__/diffusion_sdedit.cpython-39.pyc differ
 
video_to_video/diffusion/__pycache__/schedules_sdedit.cpython-39.pyc CHANGED
Binary files a/video_to_video/diffusion/__pycache__/schedules_sdedit.cpython-39.pyc and b/video_to_video/diffusion/__pycache__/schedules_sdedit.cpython-39.pyc differ
 
video_to_video/diffusion/__pycache__/solvers_sdedit.cpython-39.pyc CHANGED
Binary files a/video_to_video/diffusion/__pycache__/solvers_sdedit.cpython-39.pyc and b/video_to_video/diffusion/__pycache__/solvers_sdedit.cpython-39.pyc differ
 
video_to_video/modules/__pycache__/__init__.cpython-39.pyc CHANGED
Binary files a/video_to_video/modules/__pycache__/__init__.cpython-39.pyc and b/video_to_video/modules/__pycache__/__init__.cpython-39.pyc differ
 
video_to_video/modules/__pycache__/embedder.cpython-39.pyc CHANGED
Binary files a/video_to_video/modules/__pycache__/embedder.cpython-39.pyc and b/video_to_video/modules/__pycache__/embedder.cpython-39.pyc differ
 
video_to_video/modules/__pycache__/unet_v2v.cpython-39.pyc CHANGED
Binary files a/video_to_video/modules/__pycache__/unet_v2v.cpython-39.pyc and b/video_to_video/modules/__pycache__/unet_v2v.cpython-39.pyc differ
 
video_to_video/utils/__pycache__/__init__.cpython-39.pyc CHANGED
Binary files a/video_to_video/utils/__pycache__/__init__.cpython-39.pyc and b/video_to_video/utils/__pycache__/__init__.cpython-39.pyc differ
 
video_to_video/utils/__pycache__/config.cpython-39.pyc CHANGED
Binary files a/video_to_video/utils/__pycache__/config.cpython-39.pyc and b/video_to_video/utils/__pycache__/config.cpython-39.pyc differ
 
video_to_video/utils/__pycache__/logger.cpython-39.pyc CHANGED
Binary files a/video_to_video/utils/__pycache__/logger.cpython-39.pyc and b/video_to_video/utils/__pycache__/logger.cpython-39.pyc differ
 
video_to_video/utils/__pycache__/seed.cpython-39.pyc CHANGED
Binary files a/video_to_video/utils/__pycache__/seed.cpython-39.pyc and b/video_to_video/utils/__pycache__/seed.cpython-39.pyc differ
 
video_to_video/video_to_video_model.py CHANGED
@@ -17,10 +17,10 @@ from diffusers import AutoencoderKLTemporalDecoder
17
  import requests
18
 
19
  def download_model(url, model_path):
20
- if not os.path.exists(model_path):
21
  print(f"Model not found at {model_path}, downloading...")
22
  response = requests.get(url, stream=True)
23
- with open(model_path, 'wb') as f:
24
  for chunk in response.iter_content(chunk_size=1024):
25
  if chunk:
26
  f.write(chunk)
@@ -37,7 +37,7 @@ class VideoToVideo_sr():
37
  self.device = device # torch.device(f'cuda:0')
38
 
39
  # text_encoder
40
- text_encoder = FrozenOpenCLIPEmbedder(device=self.device, pretrained="laion2b_s32b_b79k")
41
  text_encoder.model.to(self.device)
42
  self.text_encoder = text_encoder
43
  logger.info(f'Build encoder with FrozenOpenCLIPEmbedder')
@@ -54,7 +54,7 @@ class VideoToVideo_sr():
54
  download_model(model_url, cfg.model_path)
55
 
56
  # 拼接完整路径
57
- model_file_path = os.path.join(cfg.model_path, 'heavy_deg.pt')
58
  print('model_file_path:', model_file_path)
59
 
60
  # 加载模型
@@ -80,7 +80,7 @@ class VideoToVideo_sr():
80
 
81
  # Temporal VAE
82
  vae = AutoencoderKLTemporalDecoder.from_pretrained(
83
- "stabilityai/stable-video-diffusion-img2vid", subfolder="vae", variant="fp16"
84
  )
85
  vae.eval()
86
  vae.requires_grad_(False)
 
17
  import requests
18
 
19
  def download_model(url, model_path):
20
+ if not os.path.exists(os.path.join(model_path, 'model.pt')):
21
  print(f"Model not found at {model_path}, downloading...")
22
  response = requests.get(url, stream=True)
23
+ with open(os.path.join(model_path, 'model.pt'), 'wb') as f:
24
  for chunk in response.iter_content(chunk_size=1024):
25
  if chunk:
26
  f.write(chunk)
 
37
  self.device = device # torch.device(f'cuda:0')
38
 
39
  # text_encoder
40
+ text_encoder = FrozenOpenCLIPEmbedder(device=self.device, pretrained="/home/test/Workspace/yhliu/VSR/ours/checkpoints/open_clip_pytorch_model.bin")
41
  text_encoder.model.to(self.device)
42
  self.text_encoder = text_encoder
43
  logger.info(f'Build encoder with FrozenOpenCLIPEmbedder')
 
54
  download_model(model_url, cfg.model_path)
55
 
56
  # 拼接完整路径
57
+ model_file_path = os.path.join(cfg.model_path, 'model.pt')
58
  print('model_file_path:', model_file_path)
59
 
60
  # 加载模型
 
80
 
81
  # Temporal VAE
82
  vae = AutoencoderKLTemporalDecoder.from_pretrained(
83
+ "/home/test/Workspace/yhliu/VSR/ours/checkpoints/stable-video-diffusion-img2vid", subfolder="vae", variant="fp16"
84
  )
85
  vae.eval()
86
  vae.requires_grad_(False)