Spaces:
Sleeping
Sleeping
zeeshan391
commited on
Commit
•
357c729
1
Parent(s):
3ea1132
updated
Browse files
app.py
CHANGED
@@ -28,8 +28,8 @@ callback_manager = CallbackManager([StreamingStdOutCallbackHandler()])
|
|
28 |
|
29 |
def load_model(
|
30 |
directory: str = ".",
|
31 |
-
model_name: str = "natsumura-storytelling-rp-1.0-llama-3.1-8B.
|
32 |
-
model_url: str = "https://huggingface.co/tohur/natsumura-storytelling-rp-1.0-llama-3.1-8b-GGUF/resolve/main/natsumura-storytelling-rp-1.0-llama-3.1-8B.
|
33 |
):
|
34 |
final_model_path = os.path.join(directory, model_name)
|
35 |
|
@@ -60,6 +60,7 @@ def load_model(
|
|
60 |
temperature=0.3,
|
61 |
max_tokens=2000,
|
62 |
n_ctx=1024,
|
|
|
63 |
# n_threads=8,
|
64 |
echo=False
|
65 |
)
|
|
|
28 |
|
29 |
def load_model(
|
30 |
directory: str = ".",
|
31 |
+
model_name: str = "natsumura-storytelling-rp-1.0-llama-3.1-8B.Q3_K_M.gguf",
|
32 |
+
model_url: str = "https://huggingface.co/tohur/natsumura-storytelling-rp-1.0-llama-3.1-8b-GGUF/resolve/main/natsumura-storytelling-rp-1.0-llama-3.1-8B.Q3_K_M.gguf"
|
33 |
):
|
34 |
final_model_path = os.path.join(directory, model_name)
|
35 |
|
|
|
60 |
temperature=0.3,
|
61 |
max_tokens=2000,
|
62 |
n_ctx=1024,
|
63 |
+
top_p=1,
|
64 |
# n_threads=8,
|
65 |
echo=False
|
66 |
)
|