NousResearch
/

CodeLlama-7b-hf-flash

Text Generation

text-generation-inference

Model card Files Files and versions

emozilla commited on Aug 24, 2023

Commit

80081c1

·

1 Parent(s): 0a39274

Update modeling_flash_llama.py

Files changed (1) hide show

modeling_flash_llama.py +1 -1

modeling_flash_llama.py CHANGED Viewed

@@ -31,7 +31,7 @@ from transformers.activations import ACT2FN
 from transformers.modeling_outputs import BaseModelOutputWithPast, CausalLMOutputWithPast, SequenceClassifierOutputWithPast
 from transformers.modeling_utils import PreTrainedModel
 from transformers.utils import add_start_docstrings, add_start_docstrings_to_model_forward, logging, replace_return_docstrings
-from transformers.models.llama.configuration_llama import LlamaConfig
 try:

 from transformers.modeling_outputs import BaseModelOutputWithPast, CausalLMOutputWithPast, SequenceClassifierOutputWithPast
 from transformers.modeling_utils import PreTrainedModel
 from transformers.utils import add_start_docstrings, add_start_docstrings_to_model_forward, logging, replace_return_docstrings
+from .configuration_llama import LlamaConfig
 try: