mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 17:24:32 +02:00
set default settings
This commit is contained in:
@@ -1,5 +1,12 @@
|
||||
# Change Log for SD.Next
|
||||
|
||||
## Update for 2023-08-21
|
||||
|
||||
- general:
|
||||
- all system and image paths are now relative by default
|
||||
- fix extra networks previews
|
||||
- add settings validation when performing load/save
|
||||
|
||||
## Update for 2023-08-20
|
||||
|
||||
Another release thats been baking in dev branch for a while...
|
||||
|
||||
@@ -246,7 +246,7 @@ def select_checkpoint(op='model'):
|
||||
else:
|
||||
shared.log.info("Selecting first available checkpoint")
|
||||
# shared.log.warning(f"Loading fallback checkpoint: {checkpoint_info.title}")
|
||||
shared.opts.data['sd_checkpoint'] = checkpoint_info.title
|
||||
shared.opts.data['sd_model_checkpoint'] = checkpoint_info.title
|
||||
shared.log.debug(f'Select checkpoint: {checkpoint_info.title if checkpoint_info is not None else None}')
|
||||
return checkpoint_info
|
||||
|
||||
|
||||
+10
-10
@@ -117,18 +117,18 @@ def resolve_vae(checkpoint_file):
|
||||
return None, None
|
||||
vae_near_checkpoint = find_vae_near_checkpoint(checkpoint_file)
|
||||
if vae_near_checkpoint is not None: # 3rd
|
||||
return vae_near_checkpoint, 'near checkpoint'
|
||||
return vae_near_checkpoint, 'near-checkpoint'
|
||||
if shared.opts.sd_vae == "Automatic": # 4th
|
||||
basename = os.path.splitext(os.path.basename(checkpoint_file))[0]
|
||||
if vae_dict.get(basename, None) is not None:
|
||||
return vae_dict[basename], 'in VAE dir'
|
||||
return vae_dict[basename], 'automatic'
|
||||
else:
|
||||
vae_from_options = vae_dict.get(shared.opts.sd_vae, None) # 5th
|
||||
if vae_from_options is not None:
|
||||
return vae_from_options, 'specified in settings'
|
||||
return vae_from_options, 'settings'
|
||||
vae_from_options = vae_dict.get(shared.opts.sd_vae + '.safetensors', None) # 6th
|
||||
if vae_from_options is not None:
|
||||
return vae_from_options, 'specified in settings'
|
||||
return vae_from_options, 'settings'
|
||||
shared.log.warning(f"VAE not found: {shared.opts.sd_vae}")
|
||||
return None, None
|
||||
|
||||
@@ -139,18 +139,18 @@ def load_vae_dict(filename):
|
||||
return vae_dict_1
|
||||
|
||||
|
||||
def load_vae(model, vae_file=None, vae_source="from unknown source"):
|
||||
def load_vae(model, vae_file=None, vae_source="unknown-source"):
|
||||
global loaded_vae_file # pylint: disable=global-statement
|
||||
cache_enabled = shared.opts.sd_vae_checkpoint_cache > 0
|
||||
if vae_file:
|
||||
if cache_enabled and vae_file in checkpoints_loaded:
|
||||
# use vae checkpoint cache
|
||||
shared.log.info(f"Loading VAE weights: {vae_source}: cached {get_filename(vae_file)}")
|
||||
shared.log.info(f"Loading VAE weights: {get_filename(vae_file)} source={vae_source} cached=True")
|
||||
store_base_vae(model)
|
||||
_load_vae_dict(model, checkpoints_loaded[vae_file])
|
||||
else:
|
||||
if not os.path.isfile(vae_file):
|
||||
shared.log.error(f"VAE {vae_source} doesn't exist: {vae_file}")
|
||||
shared.log.error(f"VAE doesn't exist: {vae_file} source={vae_source}")
|
||||
return
|
||||
store_base_vae(model)
|
||||
vae_dict_1 = load_vae_dict(vae_file)
|
||||
@@ -172,13 +172,13 @@ def load_vae(model, vae_file=None, vae_source="from unknown source"):
|
||||
loaded_vae_file = vae_file
|
||||
|
||||
|
||||
def load_vae_diffusers(model_file, vae_file=None, vae_source="from unknown source"):
|
||||
def load_vae_diffusers(model_file, vae_file=None, vae_source="unknown-source"):
|
||||
if vae_file is None:
|
||||
return None
|
||||
if not os.path.exists(vae_file):
|
||||
shared.log.error(f'VAE not found: {vae_file}')
|
||||
return None
|
||||
shared.log.info(f"Loading diffusers VAE: {vae_source}: {vae_file}")
|
||||
shared.log.info(f"Loading diffusers VAE: {vae_file} source={vae_source}")
|
||||
diffusers_load_config = {
|
||||
"low_cpu_mem_usage": False,
|
||||
"torch_dtype": devices.dtype_vae,
|
||||
@@ -236,7 +236,7 @@ def reload_vae_weights(sd_model=None, vae_file=unspecified):
|
||||
if vae_file == unspecified:
|
||||
vae_file, vae_source = resolve_vae(checkpoint_file)
|
||||
else:
|
||||
vae_source = "from function argument"
|
||||
vae_source = "function-argument"
|
||||
if loaded_vae_file == vae_file:
|
||||
return
|
||||
if not sd_model.has_accelerate:
|
||||
|
||||
Reference in New Issue
Block a user