Spaces:
Runtime error
Runtime error
storresbusquets
commited on
Commit
•
24a4fff
1
Parent(s):
0c7f95c
Update app.py
Browse files
app.py
CHANGED
@@ -49,7 +49,7 @@ class GradioInference:
|
|
49 |
self.tokenizer = AutoTokenizer.from_pretrained("csebuetnlp/mT5_multilingual_XLSum", truncation=True)
|
50 |
self.model = AutoModelForSeq2SeqLM.from_pretrained("csebuetnlp/mT5_multilingual_XLSum")
|
51 |
|
52 |
-
self.config = {'
|
53 |
self.llm = CTransformers(model="TheBloke/Llama-2-7B-Chat-GGML", model_file = 'llama-2-7b-chat.ggmlv3.q2_K.bin', config=self.config,callbacks=[StreamingStdOutCallbackHandler()])
|
54 |
|
55 |
|
|
|
49 |
self.tokenizer = AutoTokenizer.from_pretrained("csebuetnlp/mT5_multilingual_XLSum", truncation=True)
|
50 |
self.model = AutoModelForSeq2SeqLM.from_pretrained("csebuetnlp/mT5_multilingual_XLSum")
|
51 |
|
52 |
+
self.config = {'repetition_penalty': 1.1, 'temperature':0}
|
53 |
self.llm = CTransformers(model="TheBloke/Llama-2-7B-Chat-GGML", model_file = 'llama-2-7b-chat.ggmlv3.q2_K.bin', config=self.config,callbacks=[StreamingStdOutCallbackHandler()])
|
54 |
|
55 |
|