Allow parsing .safetensors stable diffusion model path in the metadata parser

This commit is contained in:
cmdr2 2022-12-14 10:34:36 +05:30
parent 84d606408a
commit 7dc7f70582

View File

@ -171,7 +171,7 @@ const TASK_MAPPING = {
setUI: (use_stable_diffusion_model) => {
const oldVal = stableDiffusionModelField.value
use_stable_diffusion_model = getModelPath(use_stable_diffusion_model, ['.ckpt'])
use_stable_diffusion_model = getModelPath(use_stable_diffusion_model, ['.ckpt', '.safetensors'])
stableDiffusionModelField.value = use_stable_diffusion_model
if (!stableDiffusionModelField.value) {