Spaces:
Running
on
Zero
Running
on
Zero
chore: update info
Browse files
v2.py
CHANGED
@@ -30,7 +30,7 @@ from utils import ASPECT_RATIO_OPTIONS, RATING_OPTIONS, LENGTH_OPTIONS, IDENTITY
|
|
30 |
HF_TOKEN = os.getenv("HF_TOKEN", None)
|
31 |
|
32 |
ALL_MODELS = {
|
33 |
-
"dart-v2-mixtral-160m-sft
|
34 |
"repo": "p1atdev/dart-v2-mixtral-160m-sft-8",
|
35 |
"type": "sft",
|
36 |
"class": MixtralModel,
|
@@ -180,16 +180,19 @@ class V2UI:
|
|
180 |
)
|
181 |
input_aspect_ratio = gr.Radio(
|
182 |
label="Aspect ratio",
|
|
|
183 |
choices=["ultra_wide", "wide", "square", "tall", "ultra_tall"],
|
184 |
value="tall",
|
185 |
)
|
186 |
input_length = gr.Radio(
|
187 |
label="Length",
|
|
|
188 |
choices=list(LENGTH_OPTIONS.keys()),
|
189 |
value="long",
|
190 |
)
|
191 |
input_identity = gr.Radio(
|
192 |
-
label="Keep identity
|
|
|
193 |
choices=list(IDENTITY_OPTIONS.keys()),
|
194 |
value="none",
|
195 |
)
|
@@ -197,6 +200,7 @@ class V2UI:
|
|
197 |
with gr.Accordion(label="Advanced options", open=False):
|
198 |
input_ban_tags = gr.Textbox(
|
199 |
label="Ban tags",
|
|
|
200 |
placeholder="alternate costumen, ...",
|
201 |
)
|
202 |
|
|
|
30 |
HF_TOKEN = os.getenv("HF_TOKEN", None)
|
31 |
|
32 |
ALL_MODELS = {
|
33 |
+
"dart-v2-mixtral-160m-sft": {
|
34 |
"repo": "p1atdev/dart-v2-mixtral-160m-sft-8",
|
35 |
"type": "sft",
|
36 |
"class": MixtralModel,
|
|
|
180 |
)
|
181 |
input_aspect_ratio = gr.Radio(
|
182 |
label="Aspect ratio",
|
183 |
+
info="The aspect ratio of the image.",
|
184 |
choices=["ultra_wide", "wide", "square", "tall", "ultra_tall"],
|
185 |
value="tall",
|
186 |
)
|
187 |
input_length = gr.Radio(
|
188 |
label="Length",
|
189 |
+
info="How long the total tags should be.",
|
190 |
choices=list(LENGTH_OPTIONS.keys()),
|
191 |
value="long",
|
192 |
)
|
193 |
input_identity = gr.Radio(
|
194 |
+
label="Keep identity",
|
195 |
+
info="How strict to keep the identity of the character or subject.",
|
196 |
choices=list(IDENTITY_OPTIONS.keys()),
|
197 |
value="none",
|
198 |
)
|
|
|
200 |
with gr.Accordion(label="Advanced options", open=False):
|
201 |
input_ban_tags = gr.Textbox(
|
202 |
label="Ban tags",
|
203 |
+
info="Tags to ban from the output.",
|
204 |
placeholder="alternate costumen, ...",
|
205 |
)
|
206 |
|