phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-06 02:32:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

commit

Browse source
This commit is contained in:
Botao Chen 2025-01-14 18:06:37 -08:00
parent a77b8126d0
commit db5492202f
1 changed files with 0 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/providers/inline/post_training/torchtune/recipes/lora_finetuning_single_device.py
View file

@ -587,9 +587,6 @@ class LoraFinetuningSingleDevice:
gc.collect() gc.collect()
torch.cuda.empty_cache() torch.cuda.empty_cache()
print("Allocated:", torch.cuda.memory_allocated() / 1e6, "MB")
print("Reserved: ", torch.cuda.memory_reserved() / 1e6, "MB")
return (memory_stats, checkpoints) return (memory_stats, checkpoints)
async def validation(self) -> Tuple[float, float]: async def validation(self) -> Tuple[float, float]: