zai-org
/

chatglm3-6b

Model card Files Files and versions

katuni4ka commited on Oct 21, 2024

Commit

c59cdd3

·

verified ·

1 Parent(s): 9e21dac

Update modeling_chatglm.py

Files changed (1) hide show

modeling_chatglm.py +1 -0

modeling_chatglm.py CHANGED Viewed

@@ -14,6 +14,7 @@ from torch.nn import CrossEntropyLoss, LayerNorm, MSELoss, BCEWithLogitsLoss
 from torch.nn.utils import skip_init
 from typing import Optional, Tuple, Union, List, Callable, Dict, Any
 from copy import deepcopy
 from transformers.modeling_outputs import (
     BaseModelOutputWithPast,

 from torch.nn.utils import skip_init
 from typing import Optional, Tuple, Union, List, Callable, Dict, Any
 from copy import deepcopy
+import transformers
 from transformers.modeling_outputs import (
     BaseModelOutputWithPast,