Function getModelDir

apps/node/src/modules/model-manager/model-paths.ts:117–156

(type: ModelType, save_path: string = "default")

}

export function getModelDir(type: ModelType, save_path: string = "default"): string {
  const { BASE_PATH, MODELS_DIR, FOLDER_NAMES_AND_PATHS } = getFolderNamesAndPaths();
  if (save_path !== 'default') {
    if (save_path.includes('..') || save_path.startsWith('/')) {
      console.warn(`[WARN] '${save_path}' is not allowed path. So it will be saved into 'models/etc'.`);
      return 'etc';
    } else {
      if (save_path.startsWith('custom_nodes')) {
        return `${BASE_PATH}/${save_path}`;
      } else {
        return `${MODELS_DIR}/${save_path}`;
      }
    }
  } else {
    const model_type = type;
    switch (model_type) {
      case 'checkpoints':
      case 'unclip':
        return FOLDER_NAMES_AND_PATHS.checkpoints[0][0];
      case 'VAE':
        return FOLDER_NAMES_AND_PATHS.vae[0][0];
      case 'lora':
        return FOLDER_NAMES_AND_PATHS.loras[0][0];
      case 'T2I-Adapter':
      case 'T2I-Style':
      case 'controlnet':
        return FOLDER_NAMES_AND_PATHS.controlnet[0][0];
      case 'clip_vision':
        return FOLDER_NAMES_AND_PATHS.clip_vision[0][0];
      case 'gligen':
        return FOLDER_NAMES_AND_PATHS.gligen[0][0];
      case 'upscale':
        return FOLDER_NAMES_AND_PATHS.upscale_models[0][0];
      case 'embeddings':
        return FOLDER_NAMES_AND_PATHS.embeddings[0][0];
      default:
        return 'etc';
    }
  }
}

export function getModelPath(type: string, save_path: string, file_name: string): string {
  const modelDir = getModelDir(type as any, save_path);