Merge pull request from DerThomy/main

Empty torch cache after optimizer tensor replacement
This commit is contained in:
Shakiba Kheradmand 2025-01-01 18:12:02 -08:00 committed by GitHub
commit 7b4fc9f76a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -442,6 +442,8 @@ class GaussianModel:
self._scaling = optimizable_tensors["scaling"]
self._rotation = optimizable_tensors["rotation"]
torch.cuda.empty_cache()
return optimizable_tensors