add better custom tokenizer support and tokenizer-midipiano.json
This commit is contained in:
@@ -5,6 +5,7 @@ Model = "model"
|
||||
Model_Status = "model_status"
|
||||
Model_Config = "model_config"
|
||||
Deploy_Mode = "deploy_mode"
|
||||
Midi_Vocab_Config_Type = "midi_vocab_config_type"
|
||||
|
||||
|
||||
class ModelStatus(Enum):
|
||||
@@ -13,11 +14,17 @@ class ModelStatus(Enum):
|
||||
Working = 3
|
||||
|
||||
|
||||
class MidiVocabConfig(Enum):
|
||||
Default = auto()
|
||||
Piano = auto()
|
||||
|
||||
|
||||
def init():
|
||||
global GLOBALS
|
||||
GLOBALS = {}
|
||||
set(Model_Status, ModelStatus.Offline)
|
||||
set(Deploy_Mode, False)
|
||||
set(Midi_Vocab_Config_Type, MidiVocabConfig.Default)
|
||||
|
||||
|
||||
def set(key, value):
|
||||
|
||||
Reference in New Issue
Block a user