Spaces:

shivalikasingh
/

dolly-v2-demo

Sleeping

shivi commited on Apr 20, 2023

Commit

8bd3eb9

1 Parent(s): 73ccce1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ import torch
 from transformers import pipeline
 import pandas as pd
-instruct_pipeline = pipeline(model="databricks/dolly-v2-12b", torch_dtype=torch.bfloat16, trust_remote_code=True, device_map="auto")
 def run_pipeline(prompt):

 from transformers import pipeline
 import pandas as pd
+instruct_pipeline = pipeline(model="databricks/dolly-v2-7b", torch_dtype=torch.bfloat16, trust_remote_code=True, device_map="auto")
 def run_pipeline(prompt):