Spaces:
Running
on
Zero
Running
on
Zero
Kohaku-Blueleaf
commited on
Commit
•
32aef8c
1
Parent(s):
36f7056
change some default values
Browse files
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=
|
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,
|
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
|