43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
# editable modelspaces map file
|
|
# you can add custom layout by simple syntax
|
|
# Parameter "no_linage" disables creating subdir for each model linage
|
|
# <layout_name> = PathMap(no_lineage=<no_lineage>)
|
|
# <layout_name>[<model_type>] = '<model_type_subdir>'
|
|
#
|
|
# map = {
|
|
# ... (other layouts)
|
|
# <layout_name>: "<layout_name>"
|
|
# ... (other layouts)
|
|
# }
|
|
# map must be single. Don't try to create multiple map variables. add all layouts to single map
|
|
|
|
class PathMap(dict):
|
|
def __init__(self, no_lineage: bool):
|
|
super().__init__()
|
|
self.no_lineage = no_lineage
|
|
|
|
comfy = PathMap(no_lineage=False)
|
|
comfy['audio_encoder'] = 'audio_encoders'
|
|
comfy['checkpoint'] = 'checkpoints'
|
|
comfy['clip'] = 'clip'
|
|
comfy['clip_vision'] = 'clip_vision'
|
|
comfy['controlnet'] = 'controlnet'
|
|
comfy['diffuser'] = 'diffusers'
|
|
comfy['diffusion_model'] = 'diffusion_models'
|
|
comfy['diffusion_nodel'] = 'diffusion_models'
|
|
comfy['embedding'] = 'embeddings'
|
|
comfy['gligen'] = 'gligen'
|
|
comfy['hypernetwork'] = 'hypernetworks'
|
|
comfy['lora'] = 'loras'
|
|
comfy['model_patch'] = 'model_patches'
|
|
comfy['photomaker'] = 'photomaker'
|
|
comfy['style_model'] = 'style_models'
|
|
comfy['text_encoder'] = 'text_encoders'
|
|
comfy['unet'] = 'unet'
|
|
comfy['upscale_model'] = 'upscale_models'
|
|
comfy['vae'] = 'vae'
|
|
comfy['vae_approx'] = 'vae_approx'
|
|
|
|
map = {
|
|
'comfy': comfy
|
|
} |