Update app.py
Browse files
app.py
CHANGED
@@ -49,7 +49,7 @@ class StarlingBot:
|
|
49 |
input_ids = input_ids.to(device)
|
50 |
response = model.generate(
|
51 |
input_ids=input_ids,
|
52 |
-
use_cache=
|
53 |
early_stopping=False,
|
54 |
bos_token_id=model.config.bos_token_id,
|
55 |
eos_token_id=model.config.eos_token_id,
|
|
|
49 |
input_ids = input_ids.to(device)
|
50 |
response = model.generate(
|
51 |
input_ids=input_ids,
|
52 |
+
use_cache=True,
|
53 |
early_stopping=False,
|
54 |
bos_token_id=model.config.bos_token_id,
|
55 |
eos_token_id=model.config.eos_token_id,
|