Upload adapter_layer.py
Browse files- adapter_layer.py +6 -0
adapter_layer.py
CHANGED
|
@@ -277,6 +277,12 @@ class Wildnerve_tlm01(nn.Module):
|
|
| 277 |
try:
|
| 278 |
from transformers import AutoTokenizer
|
| 279 |
self.tokenizer = AutoTokenizer.from_pretrained("gpt2", use_fast=True)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 280 |
logger.info("Initialized minimal tokenizer")
|
| 281 |
except Exception as e:
|
| 282 |
logger.error(f"Failed to initialize tokenizer: {e}")
|
|
|
|
| 277 |
try:
|
| 278 |
from transformers import AutoTokenizer
|
| 279 |
self.tokenizer = AutoTokenizer.from_pretrained("gpt2", use_fast=True)
|
| 280 |
+
|
| 281 |
+
# Fix for GPT-2 tokenizer: set pad_token to eos_token
|
| 282 |
+
if not self.tokenizer.pad_token:
|
| 283 |
+
self.tokenizer.pad_token = self.tokenizer.eos_token
|
| 284 |
+
logger.info("Set GPT-2 pad_token to eos_token")
|
| 285 |
+
|
| 286 |
logger.info("Initialized minimal tokenizer")
|
| 287 |
except Exception as e:
|
| 288 |
logger.error(f"Failed to initialize tokenizer: {e}")
|