Added a way to detach models and create them standalone
This commit is contained in:
6
Project_Model/Libs/Transformer/Enums/ModelType.py
Normal file
6
Project_Model/Libs/Transformer/Enums/ModelType.py
Normal file
@@ -0,0 +1,6 @@
|
||||
from enum import Enum, auto
|
||||
|
||||
class ModelType(Enum):
|
||||
|
||||
ENCODER_ONLY = auto()
|
||||
DECODER_ONLY = auto()
|
||||
@@ -0,0 +1 @@
|
||||
from .ModelType import ModelType
|
||||
Reference in New Issue
Block a user