fromString static method

BaseModel fromString(String value)

Converts a string to a BaseModel enum value.

Implementation

static BaseModel fromString(String value) {
  switch (value.toLowerCase()) {
    case 'sd 1.4':
    case 'sd1.4':
      return BaseModel.sd14;
    case 'sd 1.5':
    case 'sd1.5':
      return BaseModel.sd15;
    case 'sd 1.5 lcm':
    case 'sd1.5 lcm':
      return BaseModel.sd15Lcm;
    case 'sd 1.5 hyper':
    case 'sd1.5 hyper':
      return BaseModel.sd15Hyper;
    case 'sd 2.0':
    case 'sd2.0':
      return BaseModel.sd20;
    case 'sd 2.0 768':
      return BaseModel.sd20768;
    case 'sd 2.1':
    case 'sd2.1':
      return BaseModel.sd21;
    case 'sd 2.1 768':
      return BaseModel.sd21768;
    case 'sd 2.1 unclip':
      return BaseModel.sd21Unclip;
    case 'sdxl 0.9':
      return BaseModel.sdxl09;
    case 'sdxl':
    case 'sd xl':
    case 'sdxl 1.0':
      return BaseModel.sdxl10;
    case 'sdxl 1.0 lcm':
      return BaseModel.sdxl10Lcm;
    case 'sdxl distilled':
      return BaseModel.sdxlDistilled;
    case 'sdxl turbo':
    case 'sd xl turbo':
      return BaseModel.sdxlTurbo;
    case 'sdxl lightning':
    case 'sd xl lightning':
      return BaseModel.sdxlLightning;
    case 'sdxl hyper':
      return BaseModel.sdxlHyper;
    case 'sd3':
    case 'sd 3':
      return BaseModel.sd3;
    case 'sd 3.5':
      return BaseModel.sd35;
    case 'sd 3.5 medium':
      return BaseModel.sd35Medium;
    case 'sd 3.5 large':
      return BaseModel.sd35Large;
    case 'sd 3.5 large turbo':
      return BaseModel.sd35LargeTurbo;
    case 'pony':
      return BaseModel.pony;
    case 'flux.1 s':
      return BaseModel.flux1S;
    case 'flux.1 d':
      return BaseModel.flux1D;
    case 'auraflow':
      return BaseModel.auraFlow;
    case 'stable cascade':
      return BaseModel.stableCascade;
    case 'svd':
      return BaseModel.svd;
    case 'svd xt':
      return BaseModel.svdXt;
    case 'playground v2':
      return BaseModel.playgroundV2;
    case 'pixart a':
      return BaseModel.pixartA;
    case 'pixart e':
      return BaseModel.pixartE;
    case 'hunyuan 1':
      return BaseModel.hunyuan1;
    case 'hunyuan video':
      return BaseModel.hunyuanVideo;
    case 'lumina':
      return BaseModel.lumina;
    case 'kolors':
      return BaseModel.kolors;
    case 'illustrious':
      return BaseModel.illustrious;
    case 'mochi':
      return BaseModel.mochi;
    case 'ltxv':
      return BaseModel.ltxv;
    case 'cogvideox':
      return BaseModel.cogVideoX;
    case 'noobai':
      return BaseModel.noobAI;
    case 'wan video':
      return BaseModel.wanVideo;
    case 'odor':
      return BaseModel.odor;
    default:
      return BaseModel.other;
  }
}