Update app.py
Browse files
app.py
CHANGED
|
@@ -26,8 +26,8 @@ llm = Llama(model_path=filename, n_ctx=512, n_batch=126)
|
|
| 26 |
def generate_text(prompt="Who is the CEO of Apple?"):
|
| 27 |
output = llm(
|
| 28 |
prompt,
|
| 29 |
-
max_tokens=
|
| 30 |
-
temperature=0.
|
| 31 |
top_p=0.5,
|
| 32 |
echo=False,
|
| 33 |
stop=["#"],
|
|
|
|
| 26 |
def generate_text(prompt="Who is the CEO of Apple?"):
|
| 27 |
output = llm(
|
| 28 |
prompt,
|
| 29 |
+
max_tokens=1000,
|
| 30 |
+
temperature=0.5,
|
| 31 |
top_p=0.5,
|
| 32 |
echo=False,
|
| 33 |
stop=["#"],
|