Update README.md
Browse files
README.md
CHANGED
|
@@ -116,6 +116,8 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 116 |
)
|
| 117 |
model.eval()
|
| 118 |
|
|
|
|
|
|
|
| 119 |
hf_hub_download(repo_id="spow12/ChatWaifu", filename="system_dict.json", local_dir='./')
|
| 120 |
hf_hub_download(repo_id="spow12/ChatWaifu_v1.0", filename="sample_chat_history.json", local_dir='./')
|
| 121 |
|
|
|
|
| 116 |
)
|
| 117 |
model.eval()
|
| 118 |
|
| 119 |
+
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, device_map='auto')
|
| 120 |
+
|
| 121 |
hf_hub_download(repo_id="spow12/ChatWaifu", filename="system_dict.json", local_dir='./')
|
| 122 |
hf_hub_download(repo_id="spow12/ChatWaifu_v1.0", filename="sample_chat_history.json", local_dir='./')
|
| 123 |
|