peterpeter8585 commited on
Commit
d149f62
·
verified ·
1 Parent(s): aa50278

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -10
app.py CHANGED
@@ -88,17 +88,9 @@ else:
88
  #pipe=pipeline(model=m,tokenizer=t,task="text-generation")
89
  import multiprocessing
90
 
91
- from langchain_community.chat_models import ChatLlamaCpp
92
 
93
- llm = ChatLlamaCpp(
94
- temperature=0,
95
- model_path="./llama-3-open-ko-8b-instruct-preview-q5_k_m.gguf",
96
- n_ctx=10000,
97
- n_batch=300, # Should be between 1 and n_ctx, consider the amount of VRAM in your GPU.
98
- n_threads=multiprocessing.cpu_count() - 1,
99
- repeat_penalty=1.5,
100
- top_p=0.5,
101
- )
102
  from langchain.retrievers import WikipediaRetriever as Wiki
103
  import gradio as gr
104
  chatbot = gr.Chatbot(
 
88
  #pipe=pipeline(model=m,tokenizer=t,task="text-generation")
89
  import multiprocessing
90
 
91
+ from langchain_community.llms import GPT4ALL
92
 
93
+ llm = GPT4ALL(model=("./llama-3-open-ko-8b-instruct-preview-q5_k_m.gguf"))
 
 
 
 
 
 
 
 
94
  from langchain.retrievers import WikipediaRetriever as Wiki
95
  import gradio as gr
96
  chatbot = gr.Chatbot(