fix path-to-repo-parsing

This commit is contained in:
Vladimir Mandic
2024-09-15 09:01:13 -04:00
parent a6136a6eec
commit 3379232e9c
2 changed files with 7 additions and 5 deletions
+4 -3
View File
@@ -1938,8 +1938,9 @@ def remove_token_merging(sd_model):
def path_to_repo(fn: str = ''):
repo_id = fn
repo_id = repo_id.replace('Diffusers/', '').replace('Diffusers\\', '')
repo_id = repo_id.replace('diffusers/', '').replace('diffusers\\', '')
print('HERE1', fn)
repo_id = fn.replace('\\', '/').split('/')
repo_id = '/'.join(repo_id[-2:] if len(repo_id) > 1 else repo_id)
repo_id = repo_id.replace('models--', '').replace('--', '/')
print('HERE1', repo_id)
return repo_id