Spaces:
Sleeping
Sleeping
File size: 1,512 Bytes
01d5a5d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
from enum import Enum
from typing import List
class ProcessStep(Enum):
"""Training process steps"""
LIST_DOCUMENTS = "list_documents"
GENERATE_DOCUMENT_EMBEDDINGS = "generate_document_embeddings"
CHUNK_DOCUMENT = "process_chunks"
CHUNK_EMBEDDING = "chunk_embedding"
EXTRACT_DIMENSIONAL_TOPICS = "extract_dimensional_topics"
GENERATE_BIOGRAPHY = "generate_biography"
MODEL_DOWNLOAD = "model_download"
MAP_ENTITY_NETWORK = "map_your_entity_network"
DECODE_PREFERENCE_PATTERNS = "decode_preference_patterns"
REINFORCE_IDENTITY = "reinforce_identity"
AUGMENT_CONTENT_RETENTION = "augment_content_retention"
TRAIN = "train"
MERGE_WEIGHTS = "merge_weights"
CONVERT_MODEL = "convert_model"
@classmethod
def get_ordered_steps(cls) -> List["ProcessStep"]:
"""Get ordered steps"""
return [
cls.MODEL_DOWNLOAD,
cls.LIST_DOCUMENTS,
cls.GENERATE_DOCUMENT_EMBEDDINGS,
cls.CHUNK_DOCUMENT,
cls.CHUNK_EMBEDDING,
cls.EXTRACT_DIMENSIONAL_TOPICS,
cls.GENERATE_BIOGRAPHY,
cls.MAP_ENTITY_NETWORK,
cls.DECODE_PREFERENCE_PATTERNS,
cls.REINFORCE_IDENTITY,
cls.AUGMENT_CONTENT_RETENTION,
cls.TRAIN,
cls.MERGE_WEIGHTS,
cls.CONVERT_MODEL,
]
def get_method_name(self) -> str:
"""Get the corresponding method name for this step"""
return self.value
|