turing-motors
/

heron-chat-git-ja-stablelm-base-7b-v1

git_japanese_stablelm_alpha

text-generation

image-captioning

Model card Files Files and versions Community

Inoichan commited on May 2

Commit

9b3e386

•

1 Parent(s): 7093f08

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -26,7 +26,7 @@ Follow [the installation guide](https://github.com/turingmotors/heron/).
 ```python
 import torch
 from heron.models.git_llm.git_japanese_stablelm_alpha import GitJapaneseStableLMAlphaForCausalLM
-from transformers import AutoProcessor
 device_id = 0
 device = f"cuda:{device_id}"

 ```python
 import torch
 from heron.models.git_llm.git_japanese_stablelm_alpha import GitJapaneseStableLMAlphaForCausalLM
+from transformers import AutoProcessor, LlamaTokenizer
 device_id = 0
 device = f"cuda:{device_id}"