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"""
|