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