danielsteinigen
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -115,7 +115,7 @@ tokenizer = AutoTokenizer.from_pretrained(
|
|
115 |
trust_remote_code=True,
|
116 |
)
|
117 |
|
118 |
-
messages = [{"role": "User", "content": "
|
119 |
prompt_ids = tokenizer.apply_chat_template(messages, chat_template="DE", tokenize=True, add_generation_prompt=True, return_tensors="pt")
|
120 |
prediction = model.generate(
|
121 |
prompt_ids.to(model.device),
|
|
|
115 |
trust_remote_code=True,
|
116 |
)
|
117 |
|
118 |
+
messages = [{"role": "User", "content": "Hallo"}]
|
119 |
prompt_ids = tokenizer.apply_chat_template(messages, chat_template="DE", tokenize=True, add_generation_prompt=True, return_tensors="pt")
|
120 |
prediction = model.generate(
|
121 |
prompt_ids.to(model.device),
|