remove changes in textual_inversion.py

This commit is contained in:
bluelovers 2023-03-26 06:15:09 +08:00
parent fd672a79af
commit d64ff4248b

View file

@ -129,7 +129,7 @@ class EmbeddingDatabase:
if first_id not in self.ids_lookup: if first_id not in self.ids_lookup:
self.ids_lookup[first_id] = [] self.ids_lookup[first_id] = []
self.ids_lookup[first_id] = sorted(self.ids_lookup[first_id] + [(ids, embedding)], key=lambda x: (len(x[0]), x[0].casefold()), reverse=True) self.ids_lookup[first_id] = sorted(self.ids_lookup[first_id] + [(ids, embedding)], key=lambda x: len(x[0]), reverse=True)
return embedding return embedding
@ -196,7 +196,7 @@ class EmbeddingDatabase:
return return
for root, dirs, fns in os.walk(embdir.path, followlinks=True): for root, dirs, fns in os.walk(embdir.path, followlinks=True):
for fn in sorted(fns, key=str.lower): for fn in fns:
try: try:
fullfn = os.path.join(root, fn) fullfn = os.path.join(root, fn)