nazneen commited on
Commit
eda89c3
β€’
1 Parent(s): 109ed92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -17,7 +17,8 @@ instruct_pipeline_llama = pipeline(model="HuggingFaceH4/llama-7b-ift-ds-save-tes
17
  #instruct_pipeline_12b = pipeline(model="databricks/dolly-v2-12b", torch_dtype=torch.bfloat16, trust_remote_code=True, device_map="auto")
18
 
19
  def generate(query, temperature, top_p, top_k, max_new_tokens):
20
- return [instruct_pipeline_falcon(query, temperature, top_p, top_k, max_new_tokens), instruct_pipeline_llama(query, temperature, top_p, top_k, max_new_tokens)]
 
21
 
22
 
23
 
 
17
  #instruct_pipeline_12b = pipeline(model="databricks/dolly-v2-12b", torch_dtype=torch.bfloat16, trust_remote_code=True, device_map="auto")
18
 
19
  def generate(query, temperature, top_p, top_k, max_new_tokens):
20
+ return [instruct_pipeline_falcon(query, temperature=temperature, top_p=top_p, top_k=top_k, max_new_tokens=max_new_tokens),
21
+ instruct_pipeline_llama(query, temperature=temperature, top_p=top_p, top_k=top_k, max_new_tokens=max_new_tokens)]
22
 
23
 
24