szuweifu commited on
Commit
efab889
·
verified ·
1 Parent(s): b1c83b6

Upload 2 files

Browse files
Files changed (2) hide show
  1. inference.py +1 -1
  2. inference_chunk.py +1 -1
inference.py CHANGED
@@ -18,7 +18,7 @@ from utils.util import load_config, pad_or_trim_to_match
18
  from huggingface_hub import hf_hub_download
19
  RELU = nn.ReLU()
20
 
21
- config_path = hf_hub_download(repo_id="nvidia/RE-USE", filename="config.json", force_download=True)
22
 
23
  def get_filepaths(directory, file_type=None):
24
  file_paths = [] # List which will store all of the full filepaths.
 
18
  from huggingface_hub import hf_hub_download
19
  RELU = nn.ReLU()
20
 
21
+ config_path = hf_hub_download(repo_id="nvidia/RE-USE", filename="config.json")
22
 
23
  def get_filepaths(directory, file_type=None):
24
  file_paths = [] # List which will store all of the full filepaths.
inference_chunk.py CHANGED
@@ -20,7 +20,7 @@ from huggingface_hub import hf_hub_download
20
 
21
  RELU = nn.ReLU()
22
 
23
- config_path = hf_hub_download(repo_id="nvidia/RE-USE", filename="config.json", force_download=True)
24
 
25
  def get_filepaths(directory, file_type=None):
26
  file_paths = [] # List which will store all of the full filepaths.
 
20
 
21
  RELU = nn.ReLU()
22
 
23
+ config_path = hf_hub_download(repo_id="nvidia/RE-USE", filename="config.json")
24
 
25
  def get_filepaths(directory, file_type=None):
26
  file_paths = [] # List which will store all of the full filepaths.