storresbusquets commited on
Commit
24a4fff
1 Parent(s): 0c7f95c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = {'max_new_tokens': 128, '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
 
 
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