Update README.md
Browse files
README.md
CHANGED
@@ -40,7 +40,7 @@ ELYZA-japanese-Llama-2-7b + (llava-v1.5-7b - Llama-2-7b-hf)
|
|
40 |
|
41 |
## How to use the model
|
42 |
> [!WARNING]
|
43 |
-
> The code for the demo worked with 4.34.1 of transformers, but did not work properly with 4.37.2. We have not tested the code in between versions or in the latest version.<br>
|
44 |
> デモ用のコードはtransformersの4.34.1では動作しましたが、4.37.2では正常に動作しませんでした。間のバージョンや最新のバージョンでは動作確認していません。
|
45 |
|
46 |
**1. Download dependencies**
|
@@ -103,7 +103,7 @@ if __name__ == "__main__":
|
|
103 |
|
104 |
# create prompt
|
105 |
# ユーザー: <image>\n{prompt}
|
106 |
-
|
107 |
conv = conv_templates[conv_mode].copy()
|
108 |
prompt = "猫の隣には何がありますか?"
|
109 |
inp = DEFAULT_IMAGE_TOKEN + '\n' + prompt
|
|
|
40 |
|
41 |
## How to use the model
|
42 |
> [!WARNING]
|
43 |
+
> The code for the demo worked with 4.34.1 of transformers, but did not work properly with 4.37.2. We have not tested the code in between versions or in the latest version.<br><br>
|
44 |
> デモ用のコードはtransformersの4.34.1では動作しましたが、4.37.2では正常に動作しませんでした。間のバージョンや最新のバージョンでは動作確認していません。
|
45 |
|
46 |
**1. Download dependencies**
|
|
|
103 |
|
104 |
# create prompt
|
105 |
# ユーザー: <image>\n{prompt}
|
106 |
+
conv_mode = "llava_llama_2"
|
107 |
conv = conv_templates[conv_mode].copy()
|
108 |
prompt = "猫の隣には何がありますか?"
|
109 |
inp = DEFAULT_IMAGE_TOKEN + '\n' + prompt
|