Commit
·
0c5534b
1
Parent(s):
e6198e7
added accelerate
Browse files- app.py +2 -2
- pyproject.toml +1 -0
app.py
CHANGED
@@ -131,8 +131,8 @@ g = gr.Interface(
|
|
131 |
label="Output",
|
132 |
)
|
133 |
],
|
134 |
-
title="izumi-lab/
|
135 |
-
description="izumi-lab/
|
136 |
)
|
137 |
g.queue(concurrency_count=1)
|
138 |
print("loading completed")
|
|
|
131 |
label="Output",
|
132 |
)
|
133 |
],
|
134 |
+
title="izumi-lab/llama-13b-japanese-lora-v0-1ep",
|
135 |
+
description="izumi-lab/llama-13b-japanese-lora-v0-1ep is a 13B-parameter LLaMA model finetuned to follow instructions. It is trained on the [izumi-lab/llm-japanese-dataset](https://huggingface.co/datasets/izumi-lab/llm-japanese-dataset) dataset and makes use of the Huggingface llama-13b-hf implementation. For more information, please visit [the project's website](https://llm.msuzuki.me).",
|
136 |
)
|
137 |
g.queue(concurrency_count=1)
|
138 |
print("loading completed")
|
pyproject.toml
CHANGED
@@ -15,6 +15,7 @@ torch = "^2.0.1"
|
|
15 |
huggingface-hub = "^0.14.1"
|
16 |
sentencepiece = "^0.1.99"
|
17 |
bitsandbytes = "^0.38.1"
|
|
|
18 |
|
19 |
|
20 |
[tool.poetry.group.dev.dependencies]
|
|
|
15 |
huggingface-hub = "^0.14.1"
|
16 |
sentencepiece = "^0.1.99"
|
17 |
bitsandbytes = "^0.38.1"
|
18 |
+
accelerate = "^0.19.0"
|
19 |
|
20 |
|
21 |
[tool.poetry.group.dev.dependencies]
|