extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-12-24 16:08:55 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fixed model parsing...

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-10-30 23:41:26 -04:00
parent 965efc3a13
commit 83d541b60d
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
ui/media/js/dnd.js
View File

@ -112,7 +112,17 @@ const TASK_MAPPING = {
use_stable_diffusion_model: { name: 'Stable Diffusion model', use_stable_diffusion_model: { name: 'Stable Diffusion model',
setUI: (use_stable_diffusion_model) => { setUI: (use_stable_diffusion_model) => {
const oldVal = stableDiffusionModelField.value const oldVal = stableDiffusionModelField.value
const pathIdx = use_stable_diffusion_model.lastIndexOf('/')
if (pathIdx >= 0) {
use_stable_diffusion_model = use_stable_diffusion_model.slice(pathIdx + 1)
}
if (use_stable_diffusion_model.endsWith('.ckpt')) {
use_stable_diffusion_model = use_stable_diffusion_model.slice(0, use_stable_diffusion_model.length - 5)
}
stableDiffusionModelField.value = use_stable_diffusion_model stableDiffusionModelField.value = use_stable_diffusion_model
if (!stableDiffusionModelField.value) { if (!stableDiffusionModelField.value) {
stableDiffusionModelField.value = oldVal stableDiffusionModelField.value = oldVal
} }