Spaces:
Running
Running
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
|
| 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 |
|