changed import

text_encoder/text_model.py

@@ -6,7 +6,7 @@ import math
 import os
 import json
 from safetensors.torch import save_file, load_file
-from tokenizer import Tokenizer
+from tokenizer.tokenizer import Tokenizer
 
 def get_embeddings(batch_size, tokenizer, text_encoder, captions=None, neg_captions=None, device='cpu'):
     max_length = text_encoder.max_seq_length