modified: src/flux/modules/conditioner.py

Files changed (10) hide show

src/flux/__pycache__/__init__.cpython-310.pyc CHANGED Viewed

Binary files a/src/flux/__pycache__/__init__.cpython-310.pyc and b/src/flux/__pycache__/__init__.cpython-310.pyc differ

src/flux/__pycache__/_version.cpython-310.pyc CHANGED Viewed

Binary files a/src/flux/__pycache__/_version.cpython-310.pyc and b/src/flux/__pycache__/_version.cpython-310.pyc differ

src/flux/__pycache__/math.cpython-310.pyc CHANGED Viewed

Binary files a/src/flux/__pycache__/math.cpython-310.pyc and b/src/flux/__pycache__/math.cpython-310.pyc differ

src/flux/__pycache__/model.cpython-310.pyc CHANGED Viewed

Binary files a/src/flux/__pycache__/model.cpython-310.pyc and b/src/flux/__pycache__/model.cpython-310.pyc differ

src/flux/__pycache__/sampling.cpython-310.pyc CHANGED Viewed

Binary files a/src/flux/__pycache__/sampling.cpython-310.pyc and b/src/flux/__pycache__/sampling.cpython-310.pyc differ

src/flux/__pycache__/util.cpython-310.pyc CHANGED Viewed

Binary files a/src/flux/__pycache__/util.cpython-310.pyc and b/src/flux/__pycache__/util.cpython-310.pyc differ

src/flux/modules/__pycache__/autoencoder.cpython-310.pyc CHANGED Viewed

Binary files a/src/flux/modules/__pycache__/autoencoder.cpython-310.pyc and b/src/flux/modules/__pycache__/autoencoder.cpython-310.pyc differ

src/flux/modules/__pycache__/conditioner.cpython-310.pyc CHANGED Viewed

Binary files a/src/flux/modules/__pycache__/conditioner.cpython-310.pyc and b/src/flux/modules/__pycache__/conditioner.cpython-310.pyc differ

src/flux/modules/__pycache__/layers.cpython-310.pyc CHANGED Viewed

Binary files a/src/flux/modules/__pycache__/layers.cpython-310.pyc and b/src/flux/modules/__pycache__/layers.cpython-310.pyc differ

src/flux/modules/conditioner.py CHANGED Viewed

@@ -16,8 +16,8 @@ class HFEmbedder(nn.Module):
             self.hf_module: CLIPTextModel = CLIPTextModel.from_pretrained(version, **hf_kwargs)
             # --- DEBUG ---
-            print(f"CLIP Tokenizer max length: {CLIPTokenizer.model_max_length}")
-            print(f"CLIP max position embeddings: {CLIPTextModel.config.max_position_embeddings}")
         else:
             self.tokenizer: T5Tokenizer = T5Tokenizer.from_pretrained(version, max_length=max_length, truncation=True)

             self.hf_module: CLIPTextModel = CLIPTextModel.from_pretrained(version, **hf_kwargs)
             # --- DEBUG ---
+            print(f"CLIP Tokenizer max length: {self.tokenizer.model_max_length}")
+            print(f"CLIP max position embeddings: {self.hf_module.config.max_position_embeddings}")
         else:
             self.tokenizer: T5Tokenizer = T5Tokenizer.from_pretrained(version, max_length=max_length, truncation=True)