DeepSeek/DeepSeek-VL2
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepSeek-VL2 synced 2025-06-26 18:25:56 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

remove clear_cuda_cache in forward

Browse Source
This commit is contained in:
StevenLiuWen
2024-12-30 14:19:34 +08:00
parent a8341f36dd
commit 9789f97283
2 changed files with 17 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
deepseek_vl2/models/modeling_deepseek_vl_v2.py
View File

@@ -618,8 +618,6 @@ class DeepseekVLV2ForCausalLM(DeepseekVLV2PreTrainedModel):
cache_position=cache_position
)
self._clear_cuda_cache()
return outputs
def _clear_cuda_cache(self):