Spaces:
Sleeping
Sleeping
File size: 2,474 Bytes
4b722ec |
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 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 |
from src.text_generation.model_init import LLMBaseClass
class Llama3(LLMBaseClass):
"""
Initializes a Llama3 model object
"""
def __init__(self) -> None:
self.model_id = "meta-llama/Meta-Llama-3-8B"
super().__init__(self.model_id)
class Mistral(LLMBaseClass):
"""
Initializes a Mistral model object
"""
def __init__(self) -> None:
self.model_id = "mistralai/Mistral-7B-v0.3"
super().__init__(self.model_id)
class Gemma2(LLMBaseClass):
"""
Initializes a Gemma2 model object
"""
def __init__(self) -> None:
self.model_id = "google/gemma-2-9b"
super().__init__(self.model_id)
class Llama3Point1(LLMBaseClass):
"""
Initializes a Llama 3.1 object
"""
def __init__(self) -> None:
self.model_id = "meta-llama/Meta-Llama-3.1-8B"
super().__init__(self.model_id)
class Llama3Instruct(LLMBaseClass):
"""
Initializes a Llama 3 Instruct object
"""
def __init__(self) -> None:
self.model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
super().__init__(self.model_id)
class MistralInstruct(LLMBaseClass):
"""
Initializes a Mistral Instruct object
"""
def __init__(self) -> None:
self.model_id = "mistralai/Mistral-7B-Instruct-v0.1"
super().__init__(self.model_id)
class Llama3Point1Instruct(LLMBaseClass):
"""
Initializes a Llama 3.1 Instruct object
"""
def __init__(self) -> None:
self.model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
super().__init__(self.model_id)
class Phi3SmallInstruct(LLMBaseClass):
"""
Initializes a Phi3-Small-Instruct object
"""
def __init__(self) -> None:
self.model_id = "microsoft/Phi-3-small-128k-instruct"
super().__init__(self.model_id)
class GPT4(LLMBaseClass):
"""
Initializes a GPT-4 Instruct object
"""
def __init__(self) -> None:
self.model_id = "gpt-4o-mini",
super().__init__(self.model_id)
class Claude3Point5Sonnet(LLMBaseClass):
"""
Initializes a Claude3.5 Instruct object
"""
def __init__(self) -> None:
self.model_id = "claude-3-5-sonnet@20240620",
super().__init__(self.model_id)
class Gemini(LLMBaseClass):
"""
Initializes a Gemini Instruct object
"""
def __init__(self) -> None:
self.model_id = "gemini-1.0-pro",
super().__init__(self.model_id)
|