Update app.py
Browse files
app.py
CHANGED
|
@@ -62,8 +62,8 @@ def instruct_generate(
|
|
| 62 |
prompt: str = "What food do lamas eat?",
|
| 63 |
input: str = "",
|
| 64 |
max_new_tokens: int = 100,
|
| 65 |
-
top_k: int = 200,
|
| 66 |
temperature: float = 0.8,
|
|
|
|
| 67 |
) -> None:
|
| 68 |
"""Generates a response based on a given instruction and an optional input.
|
| 69 |
This script will only work with checkpoints from the instruction-tuned LLaMA-Adapter model.
|
|
|
|
| 62 |
prompt: str = "What food do lamas eat?",
|
| 63 |
input: str = "",
|
| 64 |
max_new_tokens: int = 100,
|
|
|
|
| 65 |
temperature: float = 0.8,
|
| 66 |
+
top_k: int = 200,
|
| 67 |
) -> None:
|
| 68 |
"""Generates a response based on a given instruction and an optional input.
|
| 69 |
This script will only work with checkpoints from the instruction-tuned LLaMA-Adapter model.
|