davda54 commited on
Commit
c3ac028
1 Parent(s): 62557df

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -332,7 +332,9 @@ pip install accelerate
332
  import torch
333
  from transformers import AutoTokenizer, AutoModelForCausalLM
334
 
335
- tokenizer = AutoTokenizer.from_pretrained("norallm/normistral-7b-warm")
 
 
336
 
337
  # This setup needs about 8gb VRAM
338
  # Setting `load_in_8bit=False` -> 15gb VRAM
 
332
  import torch
333
  from transformers import AutoTokenizer, AutoModelForCausalLM
334
 
335
+ tokenizer = AutoTokenizer.from_pretrained(
336
+ "norallm/normistral-7b-warm"
337
+ )
338
 
339
  # This setup needs about 8gb VRAM
340
  # Setting `load_in_8bit=False` -> 15gb VRAM