matthoffner commited on
Commit
c3fd9b2
1 Parent(s): dbd6ed3

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -10,7 +10,7 @@ from ctransformers.langchain import CTransformers
10
  from pydantic import BaseModel
11
  from typing import List, Any
12
 
13
- llm = AutoModelForCausalLM.from_pretrained("starchat-alpha-GGML",
14
  model_file="starchat-alpha-ggml-q4_0.bin",
15
  model_type="starcoder")
16
 
 
10
  from pydantic import BaseModel
11
  from typing import List, Any
12
 
13
+ llm = AutoModelForCausalLM.from_pretrained("NeoDim/starchat-alpha-GGML",
14
  model_file="starchat-alpha-ggml-q4_0.bin",
15
  model_type="starcoder")
16