Spaces:
Runtime error
Runtime error
zetavg
commited on
Commit
·
dba0ad5
1
Parent(s):
60ce564
clear cache after training
Browse files
llama_lora/ui/finetune_ui.py
CHANGED
|
@@ -447,6 +447,7 @@ Train data (first 10):
|
|
| 447 |
|
| 448 |
result_message = f"Training ended:\n{str(results)}\n\nLogs:\n{logs_str}"
|
| 449 |
print(result_message)
|
|
|
|
| 450 |
return result_message
|
| 451 |
|
| 452 |
except Exception as e:
|
|
|
|
| 447 |
|
| 448 |
result_message = f"Training ended:\n{str(results)}\n\nLogs:\n{logs_str}"
|
| 449 |
print(result_message)
|
| 450 |
+
clear_cache()
|
| 451 |
return result_message
|
| 452 |
|
| 453 |
except Exception as e:
|