7 lines
316 B
Python
7 lines
316 B
Python
# These codes are copied from modelscope revision c58451baead80d83281f063d12fb377fad415257
|
|
from modelscope.metainfo import Models
|
|
from modelscope.models.builder import MODELS
|
|
from modelscope.models.nlp.llama import \
|
|
LlamaForTextGeneration as Llama2ForTextGeneration
|
|
from modelscope.utils.constant import Tasks
|