remove duplicate section

mosaicml · Oct 16, 2023 · 0dff355 · 0dff355
1 parent e3f6657
commit 0dff355
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/diffusion/datasets/image_caption.py b/diffusion/datasets/image_caption.py
@@ -91,11 +91,6 @@ def __init__(
         else:
             self.tokenizer = AutoTokenizer.from_pretrained(tokenizer_name_or_path, subfolder='tokenizer')
 
-        if self.sdxl:
-            self.tokenizer = SDXLTokenizer(tokenizer_name_or_path)
-        else:
-            self.tokenizer = AutoTokenizer.from_pretrained(tokenizer_name_or_path, subfolder='tokenizer')
-
     def __getitem__(self, index):
         sample = super().__getitem__(index)
         out = {}

diff --git a/diffusion/evaluation/clean_fid_eval.py b/diffusion/evaluation/clean_fid_eval.py
@@ -97,7 +97,7 @@ def __init__(self,
             model=self.model,
             load_path=self.load_path,
             load_weights_only=True,
-            load_strict_model_weights=False,  # need this otherwise text encoder weights throw err
+            load_strict_model_weights=False,  # Otherwise text encoder weights throws error
             eval_dataloader=self.eval_dataloader,
             seed=self.seed)