Spaces:
Sleeping
Sleeping
matthewfarant
commited on
Commit
•
546a8cb
1
Parent(s):
3e70d24
Update app.py
Browse files
app.py
CHANGED
@@ -22,12 +22,12 @@ os.environ["GOOGLE_API_KEY"] = os.getenv('GOOGLE_API_KEY')
|
|
22 |
|
23 |
# Llama Endpoint
|
24 |
llm = HuggingFaceEndpoint(
|
25 |
-
repo_id="meta-llama/Meta-Llama-3.1-8B",
|
26 |
task="text-generation",
|
27 |
-
max_new_tokens
|
28 |
-
do_sample
|
29 |
-
repetition_penalty
|
30 |
-
|
31 |
|
32 |
llama3 = ChatHuggingFace(llm=llm, temperature = 1)
|
33 |
llama3_json = ChatHuggingFace(llm=llm, format = 'json', temperature = 0)
|
|
|
22 |
|
23 |
# Llama Endpoint
|
24 |
llm = HuggingFaceEndpoint(
|
25 |
+
repo_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
|
26 |
task="text-generation",
|
27 |
+
max_new_tokens=1000,
|
28 |
+
do_sample=False,
|
29 |
+
repetition_penalty=1.03,
|
30 |
+
)
|
31 |
|
32 |
llama3 = ChatHuggingFace(llm=llm, temperature = 1)
|
33 |
llama3_json = ChatHuggingFace(llm=llm, format = 'json', temperature = 0)
|