update
Browse files
app.py
CHANGED
@@ -43,24 +43,24 @@ model_name_mapping = {
|
|
43 |
# "flowmo_hi": "FlowMo Hi",
|
44 |
# "gpt4o": "GPT-4o",
|
45 |
"janus_pro_1b": "Janus Pro 1B/7B",
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
-
|
54 |
-
|
55 |
-
|
56 |
-
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
|
61 |
-
|
62 |
-
|
63 |
-
|
64 |
}
|
65 |
|
66 |
def load_model(model_name):
|
|
|
43 |
# "flowmo_hi": "FlowMo Hi",
|
44 |
# "gpt4o": "GPT-4o",
|
45 |
"janus_pro_1b": "Janus Pro 1B/7B",
|
46 |
+
"llamagen-ds8": "LlamaGen ds8",
|
47 |
+
"llamagen-ds16": "LlamaGen ds16",
|
48 |
+
"llamagen-ds16-t2i": "LlamaGen ds16 T2I",
|
49 |
+
"maskbit_16bit": "MaskBiT 16bit",
|
50 |
+
"maskbit_18bit": "MaskBiT 18bit",
|
51 |
+
"open_magvit2": "OpenMagViT",
|
52 |
+
"titok_b64": "Titok-b64",
|
53 |
+
"titok_bl64": "Titok-bl64",
|
54 |
+
"titok_s128": "Titok-s128",
|
55 |
+
"titok_bl128": "Titok-bl128",
|
56 |
+
"titok_l32": "Titok-l32",
|
57 |
+
"titok_sl256": "Titok-sl256",
|
58 |
+
"var_256": "VAR-256",
|
59 |
+
"var_512": "VAR-512",
|
60 |
+
"FLUX.1-dev": "FLUX.1-dev",
|
61 |
+
"infinity_d32": "Infinity-d32",
|
62 |
+
"infinity_d64": "Infinity-d64",
|
63 |
+
"bsqvit": "BSQ-VIT",
|
64 |
}
|
65 |
|
66 |
def load_model(model_name):
|