From 5f187f631fee1926a09a97f39038fce3fb449e51 Mon Sep 17 00:00:00 2001 From: Cherrytest Date: Tue, 22 Aug 2023 12:50:17 +0000 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 76b1a7e..827d5c4 100644 --- a/README.md +++ b/README.md @@ -46,7 +46,7 @@ You are trained by OpenBuddy team, (https://openbuddy.ai, https://github.com/Ope User: 睡不着怎么办? Assistant:""" -inputs = tokenizer.encode(prompt, return_tensors="pt") +inputs = tokenizer.encode(prompt, return_tensors="pt").cuda() outputs = model.generate(inputs, max_length=512) response = tokenizer.decode(outputs[0]) print(response)