Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 =
|
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 =
|
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 |
|