tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

guard on deleting secrets from env (#2653) [skip ci]

Browse Source
This commit is contained in:
Wing Lian
2025-05-12 14:18:42 -04:00
parent cf69de2eb9
commit 4780ac7c4d
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
.runpod/src/handler.py
View File

@@ -57,8 +57,10 @@ async def handler(job):
logger.info("Training Complete.") logger.info("Training Complete.")
# Cleanup # Cleanup
del os.environ["WANDB_API_KEY"] if "WANDB_API_KEY" in os.environ:
del os.environ["HF_TOKEN"] del os.environ["WANDB_API_KEY"]
if "HF_TOKEN" in os.environ:
del os.environ["HF_TOKEN"]
runpod.serverless.start({"handler": handler, "return_aggregate_stream": True}) runpod.serverless.start({"handler": handler, "return_aggregate_stream": True})