Update README.md
Browse files
README.md
CHANGED
@@ -41,8 +41,8 @@ set a seed for reproducibility:
|
|
41 |
|
42 |
```python
|
43 |
>>> from transformers import AutoTokenizer, MistralForCausalLM, set_seed
|
44 |
-
>>> tokenizer = AutoTokenizer.from_pretrained("occiglot/occiglot-7b-
|
45 |
-
>>> model = MistralForCausalLM.from_pretrained('occiglot/occiglot-7b-
|
46 |
>>> set_seed(42)
|
47 |
>>> messages = [
|
48 |
>>> {"role": "system", 'content': 'You are a helpful assistant. Please give short and concise answers.'},
|
|
|
41 |
|
42 |
```python
|
43 |
>>> from transformers import AutoTokenizer, MistralForCausalLM, set_seed
|
44 |
+
>>> tokenizer = AutoTokenizer.from_pretrained("occiglot/occiglot-7b-de-en-instruct")
|
45 |
+
>>> model = MistralForCausalLM.from_pretrained('occiglot/occiglot-7b-de-en-instruct') # You may want to use bfloat16 and/or move to GPU here
|
46 |
>>> set_seed(42)
|
47 |
>>> messages = [
|
48 |
>>> {"role": "system", 'content': 'You are a helpful assistant. Please give short and concise answers.'},
|