zmeeks commited on
Commit
b64aaf3
·
verified ·
1 Parent(s): f5e76a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -77,7 +77,7 @@ class BasicAgent:
77
  if self.model.tokenizer.pad_token is None:
78
  self.model.tokenizer.pad_token = self.model.tokenizer.eos_token
79
  # Set default padding behavior for FlashAttention
80
- self.model.tokenizer.pad_to_multiple_of = 64
81
  print("✅ Applied tokenizer padding fix for FlashAttention alignment")
82
 
83
  # If the model has a processor with tokenizer, fix that too
@@ -85,7 +85,7 @@ class BasicAgent:
85
  self.model.processor.tokenizer.padding_side = "left"
86
  if self.model.processor.tokenizer.pad_token is None:
87
  self.model.processor.tokenizer.pad_token = self.model.processor.tokenizer.eos_token
88
- self.model.processor.tokenizer.pad_to_multiple_of = 64
89
  print("✅ Applied processor tokenizer padding fix")
90
 
91
 
 
77
  if self.model.tokenizer.pad_token is None:
78
  self.model.tokenizer.pad_token = self.model.tokenizer.eos_token
79
  # Set default padding behavior for FlashAttention
80
+ self.model.tokenizer.pad_to_multiple_of = 8
81
  print("✅ Applied tokenizer padding fix for FlashAttention alignment")
82
 
83
  # If the model has a processor with tokenizer, fix that too
 
85
  self.model.processor.tokenizer.padding_side = "left"
86
  if self.model.processor.tokenizer.pad_token is None:
87
  self.model.processor.tokenizer.pad_token = self.model.processor.tokenizer.eos_token
88
+ self.model.processor.tokenizer.pad_to_multiple_of = 8
89
  print("✅ Applied processor tokenizer padding fix")
90
 
91