WildnerveAI commited on
Commit
1e02256
·
verified ·
1 Parent(s): 53d82e6

Upload adapter_layer.py

Browse files
Files changed (1) hide show
  1. 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}")