Hugging Face's logo

Spaces:
BenkHel
/
CumoThesis
Running

App Files Community
BenkHel commited on
Commit
2262f7d
·
verified ·
1 Parent(s): 2edf11e

Update cumo/model/language_model/llava_llama.py

Browse files
Files changed (1) hide show
  1. cumo/model/language_model/llava_llama.py +1 -1
cumo/model/language_model/llava_llama.py CHANGED
@@ -155,4 +155,4 @@ class LlavaLlamaForCausalLM(LlamaForCausalLM, LlavaMetaForCausalLM):
155
  return inputs
156
 
157
  AutoConfig.register("llava_llama", LlavaConfig)
158
- AutoModelForCausalLM.register(LlavaConfig, LlavaLlamaForCausalLM)
 
155
  return inputs
156
 
157
  AutoConfig.register("llava_llama", LlavaConfig)
158
+ AutoModelForCausalLM.register(LlavaConfig, LlavaLlamaForCausalLM)