xierui.0097 commited on
Commit
46518d3
·
1 Parent(s): 7d81589
video_to_video/video_to_video_model.py CHANGED
@@ -17,17 +17,10 @@ from diffusers import AutoencoderKLTemporalDecoder
17
  import requests
18
 
19
  def download_model(url, model_path):
20
- <<<<<<< HEAD
21
  if not os.path.exists(os.path.join(model_path, 'model.pt')):
22
  print(f"Model not found at {model_path}, downloading...")
23
  response = requests.get(url, stream=True)
24
  with open(os.path.join(model_path, 'model.pt'), 'wb') as f:
25
- =======
26
- if not os.path.exists(os.path.join(model_path, 'heavy_deg.pt')):
27
- print(f"Model not found at {model_path}, downloading...")
28
- response = requests.get(url, stream=True)
29
- with open(os.path.join(model_path, 'heavy_deg.pt'), 'wb') as f:
30
- >>>>>>> cf76b2c6028ef0b7cad14bee801342e5b79b8cd6
31
  for chunk in response.iter_content(chunk_size=1024):
32
  if chunk:
33
  f.write(chunk)
 
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)