DeepSpeed
33ba898d - Use get_accelerator().empty_cache() to clear GPU cache

Commit

1 year ago

Use get_accelerator().empty_cache() to clear GPU cache

References

#4804 - Inference V2 Human Eval

Author

lekurile

lekurile

Parents

Loading