"INSTALL/git@developer.sourcefind.cn:dadigang/Ventoy.git" did not exist on "fe0dda690426b8e1b73c79b0dff46e1459ea40ec"
Unverified Commit 93cb58f4 authored by Maksim Skobeev's avatar Maksim Skobeev Committed by GitHub
Browse files

Update model.py

parent 7a4e62c9
...@@ -43,7 +43,7 @@ class RealESRGAN: ...@@ -43,7 +43,7 @@ class RealESRGAN:
cache_dir = os.path.dirname(model_path) cache_dir = os.path.dirname(model_path)
local_filename = os.path.basename(model_path) local_filename = os.path.basename(model_path)
config_file_url = hf_hub_url(repo_id=config['repo_id'], filename=config['filename']) config_file_url = hf_hub_url(repo_id=config['repo_id'], filename=config['filename'])
htr = hf_hub_download(repo_id=config['repo_id'], local_dir=cache_dir, filename=config['filename']) htr = hf_hub_download(repo_id=config['repo_id'], cache_dir=cache_dir, local_dir=cache_dir, filename=config['filename'])
print(htr) print(htr)
# cached_download(config_file_url, cache_dir=cache_dir, force_filename=local_filename) # cached_download(config_file_url, cache_dir=cache_dir, force_filename=local_filename)
print('Weights downloaded to:', os.path.join(cache_dir, local_filename)) print('Weights downloaded to:', os.path.join(cache_dir, local_filename))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment