Merge pull request #771 from patriceac/patch-19

Fix restoration of models with subfolders
This commit is contained in:
cmdr2 2023-02-10 17:32:56 +05:30 committed by GitHub
commit 73abf131a6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -375,12 +375,15 @@ function readUI() {
} }
function getModelPath(filename, extensions) function getModelPath(filename, extensions)
{ {
let pathIdx = filename.lastIndexOf('/') // Linux, Mac paths let pathIdx
if (pathIdx < 0) { if (filename.includes('/models/stable-diffusion/')) {
pathIdx = filename.lastIndexOf('\\') // Windows paths. pathIdx = filename.indexOf('/models/stable-diffusion/') + 25 // Linux, Mac paths
}
else if (filename.includes('\\models\\stable-diffusion\\')) {
pathIdx = filename.indexOf('\\models\\stable-diffusion\\') + 25 // Linux, Mac paths
} }
if (pathIdx >= 0) { if (pathIdx >= 0) {
filename = filename.slice(pathIdx + 1) filename = filename.slice(pathIdx)
} }
extensions.forEach(ext => { extensions.forEach(ext => {
if (filename.endsWith(ext)) { if (filename.endsWith(ext)) {