recover attrdict

2025-02-22 05:39:07 -05:00 · 2025-02-19 19:30:58 +08:00 · 2025-02-19 19:30:58 +08:00 · b6e557c73e
commit b6e557c73e
parent cf21982aa5
2 changed files with 2 additions and 0 deletions
--- a/deepseek_vl2/models/modeling_deepseek_vl_v2.py
+++ b/deepseek_vl2/models/modeling_deepseek_vl_v2.py
@ -1,3 +1,4 @@
+from attrdict import AttrDict
 from dataclasses import dataclass
 import logging
 import gc
--- a/requirements.txt
+++ b/requirements.txt
@ -4,6 +4,7 @@ xformers>=0.0.21
 timm>=0.9.16
 accelerate
 sentencepiece
+attrdict
 einops

 # for gradio demo