Merge branch 'dev' into find_vae

1 files changed, 3 insertions, 3 deletions

diff --git a/modules/textual_inversion/dataset.py b/modules/textual_inversion/dataset.py
index af9fbcf2..b9621fc9 100644
--- a/modules/textual_inversion/dataset.py
+++ b/modules/textual_inversion/dataset.py
@@ -72,7 +72,7 @@ class PersonalizedBase(Dataset):
             except Exception:

                 continue

 

-            text_filename = os.path.splitext(path)[0] + ".txt"

+            text_filename = f"{os.path.splitext(path)[0]}.txt"

             filename = os.path.basename(path)

 

             if os.path.exists(text_filename):

@@ -118,7 +118,7 @@ class PersonalizedBase(Dataset):
                 weight = torch.ones(latent_sample.shape)

             else:

                 weight = None

-            

+

             if latent_sampling_method == "random":

                 entry = DatasetEntry(filename=path, filename_text=filename_text, latent_dist=latent_dist, weight=weight)

             else:

@@ -243,4 +243,4 @@ class BatchLoaderRandom(BatchLoader):
         return self

 

 def collate_wrapper_random(batch):

-    return BatchLoaderRandom(batch)
\ No newline at end of file
+    return BatchLoaderRandom(batch)