ZhouZJ36DL commited on
Commit
f069c07
·
1 Parent(s): 53bf18d

modified: src/flux/modules/conditioner.py

Browse files
Files changed (1) hide show
  1. src/flux/modules/conditioner.py +1 -1
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)