zeeshan391 commited on
Commit
357c729
1 Parent(s): 3ea1132
Files changed (1) hide show
  1. app.py +3 -2
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.Q4_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.Q4_K_M.gguf"
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
  )