awakenai commited on
Commit
694ef91
1 Parent(s): 82f8eb5

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -9,7 +9,7 @@ app = FastAPI()
9
 
10
  #generator = pipeline('text-generation',model='gpt2')
11
  #generator = pipeline('text-generation',model='Open-Orca/Mistral-7B-OpenOrca')
12
- generator = pipeline("text-generation", model="TheBloke/zephyr-7B-alpha-GGUF")
13
  #model = AutoModel.from_pretrained("TheBloke/zephyr-7B-alpha-GGUF")
14
 
15
 
 
9
 
10
  #generator = pipeline('text-generation',model='gpt2')
11
  #generator = pipeline('text-generation',model='Open-Orca/Mistral-7B-OpenOrca')
12
+ #generator = pipeline("text-generation", model="TheBloke/zephyr-7B-alpha-GGUF")
13
  #model = AutoModel.from_pretrained("TheBloke/zephyr-7B-alpha-GGUF")
14
 
15