cognitivess commited on
Commit
db5abb4
1 Parent(s): bbbaf23

Update cognitivess_model/modeling_cognitivess.py

Browse files
cognitivess_model/modeling_cognitivess.py CHANGED
@@ -1,4 +1,4 @@
1
- from transformers.models.llama.modeling_llama import LlamaForCausalLM
2
 
3
  class CognitivessForCausalLM(LlamaForCausalLM):
4
  config_class = CognitivessConfig
 
1
+ from transformers import LlamaForCausalLM
2
 
3
  class CognitivessForCausalLM(LlamaForCausalLM):
4
  config_class = CognitivessConfig