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 |
|