Pendrokar commited on
Commit
19b56c2
β€’
1 Parent(s): 7d90a74

manual model download

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. app.py +4 -4
  3. resources/app/python/models_manager.py +2 -1
README.md CHANGED
@@ -20,4 +20,4 @@ preload_from_hub:
20
  license: gpl-3.0
21
  ---
22
 
23
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
20
  license: gpl-3.0
21
  ---
22
 
23
+ DanRuta's xVASynth, GitHub repo: [https://github.com/DanRuta/xVA-Synth](https://github.com/DanRuta/xVA-Synth)
app.py CHANGED
@@ -36,10 +36,10 @@ def run_xvaserver():
36
 
37
  def load_model():
38
  model_name = "Pendrokar/TorchMoji"
39
- # model_path = hf_hub_download(repo_id=model_name, filename="ccby_nvidia_hifi_6670_M.pt")
40
- # model_json_path = hf_hub_download(repo_id=model_name, filename="ccby_nvidia_hifi_6670_M.json")
41
- model_path = '/tmp/hfcache/models--Pendrokar--xvapitch_nvidia_6670/snapshots/2e138a7c459fb1cb1182dd7bc66813f5325d30fd/ccby_nvidia_hifi_6670_M.pt'
42
- model_json_path = '/tmp/hfcache/models--Pendrokar--xvapitch_nvidia_6670/snapshots/2e138a7c459fb1cb1182dd7bc66813f5325d30fd/ccby_nvidia_hifi_6670_M.json'
43
  # try:
44
  # os.symlink(model_path, os.path.join('./models/ccby/', os.path.basename(model_path)))
45
  # os.symlink(model_json_path, os.path.join('./models/ccby/', os.path.basename(model_json_path)))
 
36
 
37
  def load_model():
38
  model_name = "Pendrokar/TorchMoji"
39
+ model_path = hf_hub_download(repo_id=model_name, filename="ccby_nvidia_hifi_6670_M.pt")
40
+ model_json_path = hf_hub_download(repo_id=model_name, filename="ccby_nvidia_hifi_6670_M.json")
41
+ # model_path = '/tmp/hfcache/models--Pendrokar--xvapitch_nvidia_6670/snapshots/2e138a7c459fb1cb1182dd7bc66813f5325d30fd/ccby_nvidia_hifi_6670_M.pt'
42
+ # model_json_path = '/tmp/hfcache/models--Pendrokar--xvapitch_nvidia_6670/snapshots/2e138a7c459fb1cb1182dd7bc66813f5325d30fd/ccby_nvidia_hifi_6670_M.json'
43
  # try:
44
  # os.symlink(model_path, os.path.join('./models/ccby/', os.path.basename(model_path)))
45
  # os.symlink(model_json_path, os.path.join('./models/ccby/', os.path.basename(model_json_path)))
resources/app/python/models_manager.py CHANGED
@@ -111,7 +111,8 @@ class ModelsManager(object):
111
  self.init_model(model_key, instance_index)
112
 
113
  if not os.path.exists(ckpt_path):
114
- return "ENOENT"
 
115
 
116
  if self.models_bank[model_key][instance_index].ckpt_path != ckpt_path:
117
  self.logger.info(f'ModelsManager: Loading model checkpoint: {model_key}, {ckpt_path}')
 
111
  self.init_model(model_key, instance_index)
112
 
113
  if not os.path.exists(ckpt_path):
114
+ self.logger.error('Checkpoint not found!')
115
+ raise FileNotFoundError()
116
 
117
  if self.models_bank[model_key][instance_index].ckpt_path != ckpt_path:
118
  self.logger.info(f'ModelsManager: Loading model checkpoint: {model_key}, {ckpt_path}')