Spaces:
Sleeping
Sleeping
app.py
CHANGED
|
@@ -26,7 +26,6 @@ def get_completion(prompt, model="dolly-v0-70m"):
|
|
| 26 |
)
|
| 27 |
# Examples from https://www.databricks.com/blog/2023/03/24/hello-dolly-democratizing-magic-chatgpt-open-models.html
|
| 28 |
instructions = [prompt]
|
| 29 |
-
|
| 30 |
# set some additional pipeline args
|
| 31 |
pipeline_kwargs = {'torch_dtype': "auto"}
|
| 32 |
#if gpu_family == "v100":
|
|
|
|
| 26 |
)
|
| 27 |
# Examples from https://www.databricks.com/blog/2023/03/24/hello-dolly-democratizing-magic-chatgpt-open-models.html
|
| 28 |
instructions = [prompt]
|
|
|
|
| 29 |
# set some additional pipeline args
|
| 30 |
pipeline_kwargs = {'torch_dtype': "auto"}
|
| 31 |
#if gpu_family == "v100":
|