NeMo
Safetensors
llama
mylesgoose commited on
Commit
ad7e016
1 Parent(s): 9df56ac

Upload conversation.py

Browse files
Files changed (1) hide show
  1. vision/conversation.py +2 -2
vision/conversation.py CHANGED
@@ -409,8 +409,8 @@ conv_llava_llama_3_1 = Conversation(
409
  offset=0,
410
  sep="<|eot_id|>",
411
  sep_style=SeparatorStyle.LLAMA_3_1,
412
- tokenizer_id="nvidia/Llama-3.1-Minitron-4B-Width-Base",
413
- tokenizer=safe_load_tokenizer("nvidia/Llama-3.1-Minitron-4B-Width-Base"),
414
  stop_token_ids=[128009, 128008, 128001],
415
  date_string=date.today().strftime("%d %b %Y")
416
  )
 
409
  offset=0,
410
  sep="<|eot_id|>",
411
  sep_style=SeparatorStyle.LLAMA_3_1,
412
+ tokenizer_id="mylesgoose/Llama-3.1-Minitron-4B-Width-Base",
413
+ tokenizer=safe_load_tokenizer("mylesgoose/Llama-3.1-Minitron-4B-Width-Base"),
414
  stop_token_ids=[128009, 128008, 128001],
415
  date_string=date.today().strftime("%d %b %Y")
416
  )