Spaces:

dpassov
/

NousResearch-Yarn-Mistral-7b-128k

Runtime error

dpassov commited on Jan 21, 2024

Commit

4545e10

verified ·

1 Parent(s): 238e66b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 # Use a pipeline as a high-level helper
 from transformers import pipeline
-my_pipe = pipeline("text-generation", model="TheBloke/Yarn-Mistral-7B-128k-GPTQ")
 def get_class_1(input):
     return my_pipe(input)

 # Use a pipeline as a high-level helper
 from transformers import pipeline
+my_pipe = pipeline("text-generation", model="TheBloke/Yarn-Mistral-7B-128k-GGUF")
 def get_class_1(input):
     return my_pipe(input)