tomg-group-umd
/

huginn-0125

Text Generation

Model card Files Files and versions Community

JonasGeiping commited on Jan 23

Commit

d557f58

·

verified ·

1 Parent(s): ae9243e

Update raven_modeling_minimal.py

Files changed (1) hide show

raven_modeling_minimal.py +1 -1

raven_modeling_minimal.py CHANGED Viewed

@@ -11,7 +11,7 @@ from .raven_config_minimal import RavenConfig
 from transformers.cache_utils import Cache, DynamicCache
 ###################### Huggingface Glue code I ##################################################################
-from transformers import PreTrainedModel
 from transformers.utils import ModelOutput
 from transformers.generation.utils import GenerateDecoderOnlyOutput

 from transformers.cache_utils import Cache, DynamicCache
 ###################### Huggingface Glue code I ##################################################################
+from transformers import PreTrainedModel, GenerationMixin
 from transformers.utils import ModelOutput
 from transformers.generation.utils import GenerateDecoderOnlyOutput