Skip to content

Commit cf9cbec

Browse files
TalmajTalmaj Marinc
andauthored
Reformat models variable into multiline array CORE-59 (Comfy-Org#13513)
Co-authored-by: Talmaj Marinc <talmaj@comfy.org>
1 parent 96f1cee commit cf9cbec

1 file changed

Lines changed: 82 additions & 2 deletions

File tree

comfy/supported_models.py

Lines changed: 82 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1879,6 +1879,86 @@ def get_model(self, state_dict, prefix="", device=None):
18791879
out = model_base.CogVideoX(self, image_to_video=True, device=device)
18801880
return out
18811881

1882-
models = [LotusD, Stable_Zero123, SD15_instructpix2pix, SD15, SD20, SD21UnclipL, SD21UnclipH, SDXL_instructpix2pix, SDXLRefiner, SDXL, SSD1B, KOALA_700M, KOALA_1B, Segmind_Vega, SD_X4Upscaler, Stable_Cascade_C, Stable_Cascade_B, SV3D_u, SV3D_p, SD3, StableAudio, AuraFlow, PixArtAlpha, PixArtSigma, HunyuanDiT, HunyuanDiT1, FluxInpaint, Flux, LongCatImage, FluxSchnell, GenmoMochi, LTXV, LTXAV, HunyuanVideo15_SR_Distilled, HunyuanVideo15, HunyuanImage21Refiner, HunyuanImage21, HunyuanVideoSkyreelsI2V, HunyuanVideoI2V, HunyuanVideo, CosmosT2V, CosmosI2V, CosmosT2IPredict2, CosmosI2VPredict2, ZImagePixelSpace, ZImage, Lumina2, WAN22_T2V, WAN21_T2V, WAN21_I2V, WAN21_FunControl2V, WAN21_Vace, WAN21_Camera, WAN22_Camera, WAN22_S2V, WAN21_HuMo, WAN22_Animate, WAN21_FlowRVS, WAN21_SCAIL, Hunyuan3Dv2mini, Hunyuan3Dv2, Hunyuan3Dv2_1, HiDream, Chroma, ChromaRadiance, ACEStep, ACEStep15, Omnigen2, QwenImage, Flux2, Kandinsky5Image, Kandinsky5, Anima, RT_DETR_v4, ErnieImage, SAM3, SAM31, CogVideoX_I2V, CogVideoX_T2V]
18831882

1884-
models += [SVD_img2vid]
1883+
models = [
1884+
LotusD,
1885+
Stable_Zero123,
1886+
SD15_instructpix2pix,
1887+
SD15,
1888+
SD20,
1889+
SD21UnclipL,
1890+
SD21UnclipH,
1891+
SDXL_instructpix2pix,
1892+
SDXLRefiner,
1893+
SDXL,
1894+
SSD1B,
1895+
KOALA_700M,
1896+
KOALA_1B,
1897+
Segmind_Vega,
1898+
SD_X4Upscaler,
1899+
Stable_Cascade_C,
1900+
Stable_Cascade_B,
1901+
SV3D_u,
1902+
SV3D_p,
1903+
SD3,
1904+
StableAudio,
1905+
AuraFlow,
1906+
PixArtAlpha,
1907+
PixArtSigma,
1908+
HunyuanDiT,
1909+
HunyuanDiT1,
1910+
FluxInpaint,
1911+
Flux,
1912+
LongCatImage,
1913+
FluxSchnell,
1914+
GenmoMochi,
1915+
LTXV,
1916+
LTXAV,
1917+
HunyuanVideo15_SR_Distilled,
1918+
HunyuanVideo15,
1919+
HunyuanImage21Refiner,
1920+
HunyuanImage21,
1921+
HunyuanVideoSkyreelsI2V,
1922+
HunyuanVideoI2V,
1923+
HunyuanVideo,
1924+
CosmosT2V,
1925+
CosmosI2V,
1926+
CosmosT2IPredict2,
1927+
CosmosI2VPredict2,
1928+
ZImagePixelSpace,
1929+
ZImage,
1930+
Lumina2,
1931+
WAN22_T2V,
1932+
WAN21_T2V,
1933+
WAN21_I2V,
1934+
WAN21_FunControl2V,
1935+
WAN21_Vace,
1936+
WAN21_Camera,
1937+
WAN22_Camera,
1938+
WAN22_S2V,
1939+
WAN21_HuMo,
1940+
WAN22_Animate,
1941+
WAN21_FlowRVS,
1942+
WAN21_SCAIL,
1943+
Hunyuan3Dv2mini,
1944+
Hunyuan3Dv2,
1945+
Hunyuan3Dv2_1,
1946+
HiDream,
1947+
Chroma,
1948+
ChromaRadiance,
1949+
ACEStep,
1950+
ACEStep15,
1951+
Omnigen2,
1952+
QwenImage,
1953+
Flux2,
1954+
Kandinsky5Image,
1955+
Kandinsky5,
1956+
Anima,
1957+
RT_DETR_v4,
1958+
ErnieImage,
1959+
SAM3,
1960+
SAM31,
1961+
CogVideoX_I2V,
1962+
CogVideoX_T2V,
1963+
SVD_img2vid,
1964+
]

0 commit comments

Comments
 (0)