masanorihirano commited on
Commit
0c5534b
1 Parent(s): e6198e7

added accelerate

Browse files
Files changed (2) hide show
  1. app.py +2 -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/calm-7b-lora-v0-1ep",
135
- description="izumi-lab/calm-7b-lora-v0-1ep is a 7B-parameter Calm 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 Calm-7b 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")
 
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]