zerhero commited on
Commit
884bb4f
1 Parent(s): 5b913de

add default prompt

Browse files
Files changed (4) hide show
  1. app.py +23 -25
  2. config.py +7 -7
  3. models/checkpoints.py +8 -0
  4. models/vae.py +9 -0
app.py CHANGED
@@ -29,6 +29,8 @@ from config import (
29
  DEFAULT_NEGATIVE_PROMPT,
30
  DEFAULT_POSITIVE_PROMPT
31
  )
 
 
32
 
33
  preprocessor_controlnet = {
34
  "openpose": [
@@ -183,25 +185,6 @@ os.makedirs(directory_loras, exist_ok=True)
183
  directory_vaes = 'vaes'
184
  os.makedirs(directory_vaes, exist_ok=True)
185
 
186
- # - **Download SD 1.5 Models**
187
- download_model = (
188
- "https://civitai.com/api/download/models/597138, "
189
- "https://civitai.com/api/download/models/469243, "
190
- "https://civitai.com/api/download/models/609052, "
191
- "https://civitai.com/api/download/models/5036?type=Model&format=SafeTensor&size=full&fp=fp16, "
192
- "https://civitai.com/api/download/models/574369, "
193
- "https://huggingface.co/TechnoByte/MilkyWonderland/resolve/main/milkyWonderland_v40.safetensors"
194
- )
195
- # - **Download VAEs**
196
- download_vae = (
197
- "https://civitai.com/api/download/models/128078, "
198
- "https://huggingface.co/stabilityai/sdxl-vae/resolve/main/sdxl_vae.safetensors, "
199
- "https://civitai.com/api/download/models/403131?type=VAE&format=SafeTensor, "
200
- "https://huggingface.co/nubby/blessed-sdxl-vae-fp16-fix/resolve/main/sdxl_vae-fp16fix-c-1.1-b-0.5.safetensors?download=true, "
201
- "https://huggingface.co/nubby/blessed-sdxl-vae-fp16-fix/resolve/main/sdxl_vae-fp16fix-blessed.safetensors?download=true, "
202
- "https://huggingface.co/digiplay/VAE/resolve/main/vividReal_v20.safetensors?download=true, "
203
- "https://huggingface.co/fp16-guy/anything_kl-f8-anime2_vae-ft-mse-840000-ema-pruned_blessed_clearvae_fp16_cleaned/resolve/main/vae-ft-mse-840000-ema-pruned_fp16.safetensors?download=true"
204
- )
205
  # - **Download LoRAs**
206
  download_lora = (
207
  "https://civitai.com/api/download/models/423719, "
@@ -253,8 +236,8 @@ load_diffusers_format_model = [
253
  'rubbrband/realcartoonRealistic_v14',
254
  ]
255
 
256
- CIVITAI_API_KEY = os.environ.get("CIVITAI_API_KEY")
257
- hf_token = os.environ.get("HF_TOKEN")
258
 
259
  # Download stuffs
260
  for url in [url.strip() for url in download_model.split(',')]:
@@ -1162,7 +1145,12 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
1162
  with gr.Accordion("IP-Adapter", open=False, visible=True): ##############
1163
 
1164
  IP_MODELS = sorted(list(set(IP_ADAPTERS_SD + IP_ADAPTERS_SDXL)))
1165
- MODE_IP_OPTIONS = ["original", "style", "layout", "style+layout"]
 
 
 
 
 
1166
 
1167
  with gr.Accordion("IP-Adapter 1", open=False, visible=True):
1168
  image_ip1 = gr.Image(label="IP Image", type="filepath")
@@ -1178,10 +1166,20 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
1178
  scale_ip2 = gr.Slider(minimum=0., maximum=2., step=0.01, value=0.7, label="Scale")
1179
 
1180
  with gr.Accordion("ControlNet / Img2img / Inpaint", open=False, visible=True):
1181
- image_control = gr.Image(label="Image ControlNet/Inpaint/Img2img", type="filepath")
1182
- image_mask_gui = gr.Image(label="Image Mask", type="filepath")
 
 
 
 
 
 
1183
  strength_gui = gr.Slider(
1184
- minimum=0.01, maximum=1.0, step=0.01, value=0.55, label="Strength",
 
 
 
 
1185
  info="This option adjusts the level of changes for img2img and inpainting."
1186
  )
1187
  image_resolution_gui = gr.Slider(
 
29
  DEFAULT_NEGATIVE_PROMPT,
30
  DEFAULT_POSITIVE_PROMPT
31
  )
32
+ from models.vae import VAE_LIST as download_vae
33
+ from models.checkpoints import CHECKPOINT_LIST as download_model
34
 
35
  preprocessor_controlnet = {
36
  "openpose": [
 
185
  directory_vaes = 'vaes'
186
  os.makedirs(directory_vaes, exist_ok=True)
187
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
188
  # - **Download LoRAs**
189
  download_lora = (
190
  "https://civitai.com/api/download/models/423719, "
 
236
  'rubbrband/realcartoonRealistic_v14',
237
  ]
238
 
239
+ CIVITAI_API_KEY: str = os.environ.get("CIVITAI_API_KEY")
240
+ hf_token: str = os.environ.get("HF_TOKEN")
241
 
242
  # Download stuffs
243
  for url in [url.strip() for url in download_model.split(',')]:
 
1145
  with gr.Accordion("IP-Adapter", open=False, visible=True): ##############
1146
 
1147
  IP_MODELS = sorted(list(set(IP_ADAPTERS_SD + IP_ADAPTERS_SDXL)))
1148
+ MODE_IP_OPTIONS = [
1149
+ "original",
1150
+ "style",
1151
+ "layout",
1152
+ "style+layout"
1153
+ ]
1154
 
1155
  with gr.Accordion("IP-Adapter 1", open=False, visible=True):
1156
  image_ip1 = gr.Image(label="IP Image", type="filepath")
 
1166
  scale_ip2 = gr.Slider(minimum=0., maximum=2., step=0.01, value=0.7, label="Scale")
1167
 
1168
  with gr.Accordion("ControlNet / Img2img / Inpaint", open=False, visible=True):
1169
+ image_control = gr.Image(
1170
+ label="Image ControlNet/Inpaint/Img2img",
1171
+ type="filepath"
1172
+ )
1173
+ image_mask_gui = gr.Image(
1174
+ label="Image Mask",
1175
+ type="filepath"
1176
+ )
1177
  strength_gui = gr.Slider(
1178
+ minimum=0.01,
1179
+ maximum=1.0,
1180
+ step=0.01,
1181
+ value=0.55,
1182
+ label="Strength",
1183
  info="This option adjusts the level of changes for img2img and inpainting."
1184
  )
1185
  image_resolution_gui = gr.Slider(
config.py CHANGED
@@ -1,12 +1,7 @@
1
  MINIMUM_IMAGE_NUMBER = 1
2
  MAXIMUM_IMAGE_NUMBER = 10
3
- DEFAULT_NEGATIVE_PROMPT = """(EasyNegative:1.05), easynegative, bad_prompt_version2, (poorly rendered), ugly, disfigured,
4
- cross eyed, cloned face, bad symmetry, bad anatomy, low quality, blurry, text, watermark, logo,
5
- signature, jpeg, artifacts, monochrome, paintings, oil, (hands:1.15), European Woman, woman,
6
- noise, dark skin, (3d), By bad artist -neg,bhands-neg, canvas frame, (badly drawn), (badly painted),
7
- """
8
 
9
- DEFAULT_POSITIVE_PROMPT = """1girl, loli, serious, cocky, sporty, basketball jersey, lakers, basketball, half body,
10
  (basketball court), sweaty, dribble, high detailed, sunny, day light,
11
 
12
  score_9, score_8_up, score_7_up, very aesthetic,
@@ -26,5 +21,10 @@ five fingers,
26
  two hands,
27
  short girl, narrow body, detailed face, petite,
28
  medium boobs,
29
- armpits, (naval),
 
 
 
 
 
30
  """
 
1
  MINIMUM_IMAGE_NUMBER = 1
2
  MAXIMUM_IMAGE_NUMBER = 10
 
 
 
 
 
3
 
4
+ DEFAULT_POSITIVE_PROMPT = """1girl, serious, cocky, sporty, basketball jersey, lakers, basketball, half body,
5
  (basketball court), sweaty, dribble, high detailed, sunny, day light,
6
 
7
  score_9, score_8_up, score_7_up, very aesthetic,
 
21
  two hands,
22
  short girl, narrow body, detailed face, petite,
23
  medium boobs,
24
+ (armpits, (naval), crop top), outdoor
25
+ """
26
+ DEFAULT_NEGATIVE_PROMPT = """(EasyNegative:1.05), easynegative, bad_prompt_version2, (poorly rendered), ugly, disfigured,
27
+ cross eyed, cloned face, bad symmetry, bad anatomy, low quality, blurry, text, watermark, logo,
28
+ signature, jpeg, artifacts, monochrome, paintings, oil, (hands:1.15), European Woman, woman,
29
+ noise, dark skin, (3d), By bad artist -neg,bhands-neg, canvas frame, (badly drawn), (badly painted),
30
  """
models/checkpoints.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ CHECKPOINT_LIST = (
2
+ "https://civitai.com/api/download/models/597138, "
3
+ "https://civitai.com/api/download/models/469243, "
4
+ "https://civitai.com/api/download/models/609052, "
5
+ "https://civitai.com/api/download/models/5036?type=Model&format=SafeTensor&size=full&fp=fp16, "
6
+ "https://civitai.com/api/download/models/574369, "
7
+ "https://huggingface.co/TechnoByte/MilkyWonderland/resolve/main/milkyWonderland_v40.safetensors"
8
+ )
models/vae.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ VAE_LIST = (
2
+ "https://civitai.com/api/download/models/128078, "
3
+ "https://huggingface.co/stabilityai/sdxl-vae/resolve/main/sdxl_vae.safetensors, "
4
+ "https://civitai.com/api/download/models/403131?type=VAE&format=SafeTensor, "
5
+ "https://huggingface.co/nubby/blessed-sdxl-vae-fp16-fix/resolve/main/sdxl_vae-fp16fix-c-1.1-b-0.5.safetensors?download=true, "
6
+ "https://huggingface.co/nubby/blessed-sdxl-vae-fp16-fix/resolve/main/sdxl_vae-fp16fix-blessed.safetensors?download=true, "
7
+ "https://huggingface.co/digiplay/VAE/resolve/main/vividReal_v20.safetensors?download=true, "
8
+ "https://huggingface.co/fp16-guy/anything_kl-f8-anime2_vae-ft-mse-840000-ema-pruned_blessed_clearvae_fp16_cleaned/resolve/main/vae-ft-mse-840000-ema-pruned_fp16.safetensors?download=true"
9
+ )