baichuan-inc
/

Baichuan-13B-Base

Text Generation

text-generation-inference

Model card Files Files and versions Community

hiyouga commited on Jul 15, 2023

Commit

3a663f5

•

1 Parent(s): 1f41c4b

Update modeling_baichuan.py

Files changed (1) hide show

modeling_baichuan.py +1 -0

modeling_baichuan.py CHANGED Viewed

@@ -12,6 +12,7 @@ from transformers import PreTrainedModel
 from transformers.activations import ACT2FN
 from transformers.modeling_outputs import BaseModelOutputWithPast, CausalLMOutputWithPast
 from transformers.utils import logging
 from .configuration_baichuan import BaichuanConfig

 from transformers.activations import ACT2FN
 from transformers.modeling_outputs import BaseModelOutputWithPast, CausalLMOutputWithPast
 from transformers.utils import logging
+from transformers.generation.utils import GenerationConfig
 from .configuration_baichuan import BaichuanConfig