zmeeks commited on
Commit
75abe69
·
verified ·
1 Parent(s): 7ca1a6f

Update app.py

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