remove hardcoded subfolder name from model downloader

This commit is contained in:
Lincoln Stein 2023-10-08 16:43:20 -04:00 committed by Kent Keirsey
parent a681fa4b03
commit fe0cf2c160

View File

@ -335,7 +335,7 @@ class ModelInstall(object):
# list all the files in the repo # list all the files in the repo
files = [x.rfilename for x in hinfo.siblings] files = [x.rfilename for x in hinfo.siblings]
if subfolder: if subfolder:
files = [x for x in files if x.startswith("v2/")] files = [x for x in files if x.startswith(f"{subfolder}/")]
prefix = f"{subfolder}/" if subfolder else "" prefix = f"{subfolder}/" if subfolder else ""
location = None location = None