#   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['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['test_encoder'] = 'text_encoders'
comfy['unet'] = 'unet'
comfy['upscale_model'] = 'upscale_models'
comfy['vae'] = 'vae'
comfy['vae_approx'] = 'vae_approx'

map = {
    'comfy': comfy
}