Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,10 @@ class LlamaInterface:
|
|
38 |
"max_length": 512,
|
39 |
"temperature": 0.01,
|
40 |
"do_sample": True,
|
41 |
-
"
|
|
|
|
|
|
|
42 |
"top_p": 0.95,
|
43 |
}
|
44 |
|
|
|
38 |
"max_length": 512,
|
39 |
"temperature": 0.01,
|
40 |
"do_sample": True,
|
41 |
+
"pad_token_id": self.tokenizer.pad_token_id,
|
42 |
+
"eos_token_id": self.tokenizer.eos_token_id,
|
43 |
+
"num_return_sequences": 1,
|
44 |
+
"top_k": 50,
|
45 |
"top_p": 0.95,
|
46 |
}
|
47 |
|