matthoffner commited on
Commit
b8ca88a
ยท
1 Parent(s): cc30c57

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +0 -10
main.py CHANGED
@@ -9,16 +9,6 @@ from ctransformers import AutoModelForCausalLM
9
  from pydantic import BaseModel
10
  from sse_starlette.sse import EventSourceResponse
11
 
12
- config = {
13
- "temperature": 0.8,
14
- "top_p": 0.95,
15
- "top_k": 50,
16
- "max_new_tokens": 1024,
17
- "use_cache": True,
18
- "do_sample": True,
19
- "repetition_penalty": 1.02,
20
- "max_seq_len": 4096
21
- }
22
  llm = AutoModelForCausalLM.from_pretrained('TheBloke/MPT-7B-Storywriter-GGML',
23
  model_file='mpt-7b-storywriter.ggmlv3.q4_0.bin',
24
  model_type='mpt')
 
9
  from pydantic import BaseModel
10
  from sse_starlette.sse import EventSourceResponse
11
 
 
 
 
 
 
 
 
 
 
 
12
  llm = AutoModelForCausalLM.from_pretrained('TheBloke/MPT-7B-Storywriter-GGML',
13
  model_file='mpt-7b-storywriter.ggmlv3.q4_0.bin',
14
  model_type='mpt')