EQ-bench / EQ-Bench

A benchmark for emotional intelligence in large language models
MIT License
195 stars 17 forks source link

Fix GPU cleanup #30

Closed djstrong closed 4 months ago

djstrong commented 4 months ago

Fixes https://github.com/EQ-bench/EQ-Bench/issues/29

After running benchmark memory was not freed: GPU Memory after (23894294528, 42285268992) Now it is free: GPU Memory after (41722183680, 42285268992)