xlr8 commited on
Commit
1c78082
·
1 Parent(s): 891eb83

update to bf16 model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -75,7 +75,7 @@ def generate_response(message, history, initial_prompt, user_role, assistant_rol
75
  full_prompt = initial_prompt + "\n\n" + formatted_input + f"<{assistant_role}>:"
76
 
77
  completion = client.completions.create(
78
- model="meta-llama/Meta-Llama-3.1-405B-FP8",
79
  prompt=full_prompt,
80
  temperature=0.7,
81
  frequency_penalty=0.1,
 
75
  full_prompt = initial_prompt + "\n\n" + formatted_input + f"<{assistant_role}>:"
76
 
77
  completion = client.completions.create(
78
+ model="meta-llama/Meta-Llama-3.1-405B",
79
  prompt=full_prompt,
80
  temperature=0.7,
81
  frequency_penalty=0.1,