Kohaku-Blueleaf commited on
Commit
32aef8c
1 Parent(s): 36f7056

change some default values

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -148,7 +148,7 @@ def encode_cropped_prompt_77tokens(txt: str):
148
 
149
 
150
  @torch.inference_mode()
151
- def encode_cropped_prompt(txt: str, max_length=225):
152
  cond_ids = tokenizer(
153
  txt,
154
  padding="max_length",
@@ -202,7 +202,7 @@ def numpy2pytorch(imgs):
202
  @torch.inference_mode()
203
  def interrogator_process(x):
204
  img = Image.fromarray(x)
205
- rating, features, chars = get_wd14_tags(img, general_threshold=0.3, no_underline=True)
206
  result = ""
207
  for char in chars:
208
  result += char
 
148
 
149
 
150
  @torch.inference_mode()
151
+ def encode_cropped_prompt(txt: str, max_length=150):
152
  cond_ids = tokenizer(
153
  txt,
154
  padding="max_length",
 
202
  @torch.inference_mode()
203
  def interrogator_process(x):
204
  img = Image.fromarray(x)
205
+ rating, features, chars = get_wd14_tags(img, no_underline=True)
206
  result = ""
207
  for char in chars:
208
  result += char