Spaces:
Runtime error
Runtime error
tonic
commited on
Commit
Β·
2fdb696
1
Parent(s):
394eed4
refactor
Browse files
prometheus/model/__init__.py
CHANGED
@@ -1,2 +1,2 @@
|
|
1 |
-
from .language_model.
|
2 |
-
from .language_model.
|
|
|
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 ..
|
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.
|
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
|