Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
f069c07
1
Parent(s):
53bf18d
modified: src/flux/modules/conditioner.py
Browse files
src/flux/modules/conditioner.py
CHANGED
|
@@ -12,7 +12,7 @@ class HFEmbedder(nn.Module):
|
|
| 12 |
|
| 13 |
if self.is_clip:
|
| 14 |
#self.tokenizer: CLIPTokenizer = CLIPTokenizer.from_pretrained(version, max_length=max_length)
|
| 15 |
-
self.tokenizer: CLIPTokenizer = CLIPTokenizer.from_pretrained("", max_length=max_length)
|
| 16 |
self.hf_module: CLIPTextModel = CLIPTextModel.from_pretrained(version, **hf_kwargs)
|
| 17 |
else:
|
| 18 |
self.tokenizer: T5Tokenizer = T5Tokenizer.from_pretrained(version, max_length=max_length)
|
|
|
|
| 12 |
|
| 13 |
if self.is_clip:
|
| 14 |
#self.tokenizer: CLIPTokenizer = CLIPTokenizer.from_pretrained(version, max_length=max_length)
|
| 15 |
+
self.tokenizer: CLIPTokenizer = CLIPTokenizer.from_pretrained("/home/user/app/models/tokenizer", max_length=max_length)
|
| 16 |
self.hf_module: CLIPTextModel = CLIPTextModel.from_pretrained(version, **hf_kwargs)
|
| 17 |
else:
|
| 18 |
self.tokenizer: T5Tokenizer = T5Tokenizer.from_pretrained(version, max_length=max_length)
|