tonic commited on
Commit
2fdb696
β€’
1 Parent(s): 394eed4
prometheus/model/__init__.py CHANGED
@@ -1,2 +1,2 @@
1
- from .language_model.llava_llama import LlavaLlamaForCausalLM, LlavaConfig
2
- from .language_model.llava_mpt import LlavaMPTForCausalLM, LlavaMPTConfig
 
1
+ from .language_model.prometheusllll import LlavaLlamaForCausalLM, LlavaConfig
2
+ from .language_model.prometheusllmpt import LlavaMPTForCausalLM, LlavaMPTConfig
prometheus/model/language_model/{llava_llama.py β†’ prometheusllll.py} RENAMED
@@ -23,7 +23,7 @@ from transformers import AutoConfig, AutoModelForCausalLM, \
23
 
24
  from transformers.modeling_outputs import CausalLMOutputWithPast
25
 
26
- from ..llava_arch import LlavaMetaModel, LlavaMetaForCausalLM
27
 
28
 
29
  class LlavaConfig(LlamaConfig):
 
23
 
24
  from transformers.modeling_outputs import CausalLMOutputWithPast
25
 
26
+ from ..prometheus_arch import LlavaMetaModel, LlavaMetaForCausalLM
27
 
28
 
29
  class LlavaConfig(LlamaConfig):
prometheus/model/language_model/{llava_mpt.py β†’ prometheusllmpt.py} RENAMED
@@ -24,7 +24,7 @@ from transformers import AutoConfig, AutoModelForCausalLM
24
  from transformers.modeling_outputs import CausalLMOutputWithPast
25
 
26
  from .mpt.modeling_mpt import MPTConfig, MPTForCausalLM, MPTModel
27
- from prometheus.model.llava_arch import LlavaMetaModel, LlavaMetaForCausalLM
28
 
29
 
30
  class LlavaMPTConfig(MPTConfig):
 
24
  from transformers.modeling_outputs import CausalLMOutputWithPast
25
 
26
  from .mpt.modeling_mpt import MPTConfig, MPTForCausalLM, MPTModel
27
+ from prometheus.model.prometheus_arch import LlavaMetaModel, LlavaMetaForCausalLM
28
 
29
 
30
  class LlavaMPTConfig(MPTConfig):
prometheus/model/{llava_arch.py β†’ prometheus_arch.py} RENAMED
File without changes