Skip to content

Commit

Permalink
fix formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
BetaDoggo authored and Acly committed Jan 16, 2025
1 parent e424897 commit 5fddaaa
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions ai_diffusion/ui/autocomplete.py
Original file line number Diff line number Diff line change
Expand Up @@ -176,11 +176,11 @@ def _reload_tag_model(self):
with tag_path.open("r", encoding="utf-8") as f:
csv_reader = csv.reader(f)
for tag, type_str, count, _aliases in csv_reader:
if type_str.isdigit(): # skip header rows if they exist
if type_str.isdigit(): # skip header rows if they exist
tag = tag.replace("_", " ")
try:
tag_type = TagType(int(type_str))
except: # default to general category if category is not recognised
except: # default to general category if category is not recognised
tag_type = TagType(0)
count = int(count)
count_str = str(count)
Expand Down

0 comments on commit 5fddaaa

Please sign in to comment.