lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-28 10:11:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

add data-dir flag and set all user data directories based on it

Browse Source
This commit is contained in:
Max Audron
2023-01-25 17:15:42 +01:00
parent 9beb794e0b
commit 5eee2ac398
14 changed files with 39 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
modules/sd_vae.py
View File

@@ -3,13 +3,12 @@ import safetensors.torch
import os
import collections
from collections import namedtuple
from modules import shared, devices, script_callbacks, sd_models
from modules.paths import models_path
from modules import paths, shared, devices, script_callbacks, sd_models
import glob
from copy import deepcopy
vae_path = os.path.abspath(os.path.join(models_path, "VAE"))
vae_path = os.path.abspath(os.path.join(paths.models_path, "VAE"))
vae_ignore_keys = {"model_ema.decay", "model_ema.num_updates"}
vae_dict = {}