refactor tokenizer

This commit is contained in:
Paul Gauthier 2024-04-19 12:08:35 -07:00
parent 6f1cebc4c2
commit 547ae142ba
3 changed files with 4 additions and 9 deletions

View file

@ -52,7 +52,7 @@ class RepoMap:
self.max_map_tokens = map_tokens
self.tokenizer = main_model.tokenizer
self.token_count = main_model.token_count
self.repo_content_prefix = repo_content_prefix
def get_repo_map(self, chat_files, other_files):
@ -89,9 +89,6 @@ class RepoMap:
return repo_content
def token_count(self, string):
return len(self.tokenizer(string))
def get_rel_fname(self, fname):
return os.path.relpath(fname, self.root)