Update app.py
Browse files
app.py
CHANGED
@@ -14,10 +14,10 @@ model_name = "bn22/Mistral-7B-Instruct-v0.1-sharded"
|
|
14 |
# function for loading 4-bit quantized model
|
15 |
def load_quantized_model(model_name: str):
|
16 |
|
17 |
-
model =
|
18 |
-
repo_id="
|
19 |
-
model_kwargs={"temperature":0.1,
|
20 |
-
|
21 |
|
22 |
"""
|
23 |
:param model_name: Name or path of the model to be loaded.
|
|
|
14 |
# function for loading 4-bit quantized model
|
15 |
def load_quantized_model(model_name: str):
|
16 |
|
17 |
+
model = HuggingFaceHub(
|
18 |
+
repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
|
19 |
+
model_kwargs={"max_length": 1048, "temperature":0.2, "max_new_tokens":256, "top_p":0.95, "repetition_penalty":1.0},
|
20 |
+
)
|
21 |
|
22 |
"""
|
23 |
:param model_name: Name or path of the model to be loaded.
|