alihan/SwarmUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

autoconvert civit green urls

Browse Source 
for #252
This commit is contained in:
Alex "mcmonkey" Goodwin
2024-09-14 13:18:25 +09:00
parent da96942ef8
commit 3526dab56d
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/wwwroot/js/genpage/utiltab.js
View File

@@ -188,6 +188,7 @@ class ModelDownloaderUtil {
this.folders = getRequiredElementById('model_downloader_folder');
this.hfPrefix = 'https://huggingface.co/';
this.civitPrefix = 'https://civitai.com/';
this.civitGreenPrefix = 'https://civitai.green/';
}
reloadFolders() {
@@ -284,6 +285,9 @@ class ModelDownloaderUtil {
}
parseCivitaiUrl(url) {
if (url.startsWith(this.civitGreenPrefix)) {
url = this.civitPrefix + url.substring(this.civitGreenPrefix.length);
}
let parts = url.substring(this.civitPrefix.length).split('/', 4); // 'models', id, name + sometimes version OR 'api', 'download', 'models', versid
if (parts.length == 2 && parts[0] == 'models' && parts[1].includes('?')) {
let subparts = parts[1].split('?', 2);
@@ -350,6 +354,9 @@ class ModelDownloaderUtil {
this.button.disabled = false;
return;
}
if (url.startsWith(this.civitGreenPrefix)) {
url = this.civitPrefix + url.substring(this.civitGreenPrefix.length);
}
if (url.startsWith(this.civitPrefix)) {
let parts = url.substring(this.civitPrefix.length).split('/', 4); // 'models', id, name + sometimes version OR 'api', 'download', 'models', versid
if (parts.length == 2 && parts[0] == 'models' && parts[1].includes('?')) {