Spaces:

matthoffner
/

starchat-ggml

Paused

matthoffner commited on Jun 11, 2023

Commit

f116849

•

1 Parent(s): dceb191

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -10,7 +10,7 @@ from pydantic import BaseModel, Field
 from typing_extensions import Literal
 from dialogue import DialogueTemplate
-llm = AutoModelForCausalLM.from_pretrained("TheBloke/starchat-alpha-GGML",
                                            model_file="starchat-beta.ggmlv3.q4_0.bin",
                                            model_type="starcoder")

 from typing_extensions import Literal
 from dialogue import DialogueTemplate
+llm = AutoModelForCausalLM.from_pretrained("TheBloke/starchat-beta-GGML",
                                            model_file="starchat-beta.ggmlv3.q4_0.bin",
                                            model_type="starcoder")