TeePoat commited on
Commit
cab1505
·
verified ·
1 Parent(s): 24bcbb2

Update models/transformer/text_generator.py

Browse files
models/transformer/text_generator.py CHANGED
@@ -11,7 +11,7 @@ class TextGenerator:
11
  Загружаем модель и токенизатор из указанного пути.
12
  """
13
  model_path = Path(data_path) / model_name
14
- self.tokenizer = modified_tokenizer(model_path, None, data_path, use_auth_token=HF_SECRET)
15
  self.model = GPT2LMHeadModel.from_pretrained(str(model_path), device_map="auto")
16
  self.model.eval()
17
 
 
11
  Загружаем модель и токенизатор из указанного пути.
12
  """
13
  model_path = Path(data_path) / model_name
14
+ self.tokenizer = modified_tokenizer(model_path, None, data_path)
15
  self.model = GPT2LMHeadModel.from_pretrained(str(model_path), device_map="auto")
16
  self.model.eval()
17