Spaces:
Running
Running
add error handling for monitoring
Browse files
model.py
CHANGED
|
@@ -172,7 +172,17 @@ class SmolLM3Model:
|
|
| 172 |
# Override with kwargs
|
| 173 |
training_args.update(kwargs)
|
| 174 |
|
| 175 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 176 |
|
| 177 |
def save_pretrained(self, path: str):
|
| 178 |
"""Save model and tokenizer"""
|
|
|
|
| 172 |
# Override with kwargs
|
| 173 |
training_args.update(kwargs)
|
| 174 |
|
| 175 |
+
# Debug: Check for any boolean values that might be causing issues
|
| 176 |
+
for key, value in training_args.items():
|
| 177 |
+
if isinstance(value, bool):
|
| 178 |
+
logger.info(f"Boolean argument: {key} = {value}")
|
| 179 |
+
|
| 180 |
+
try:
|
| 181 |
+
return TrainingArguments(**training_args)
|
| 182 |
+
except Exception as e:
|
| 183 |
+
logger.error(f"Failed to create TrainingArguments: {e}")
|
| 184 |
+
logger.error(f"Training arguments: {training_args}")
|
| 185 |
+
raise
|
| 186 |
|
| 187 |
def save_pretrained(self, path: str):
|
| 188 |
"""Save model and tokenizer"""
|