Update README.md
Browse files
    	
        README.md
    CHANGED
    
    | 
         @@ -25,11 +25,7 @@ The following contains a code snippet illustrating how to use the model generate 
     | 
|
| 25 | 
         
             
            ```python
         
     | 
| 26 | 
         
             
            from transformers import AutoModelForCausalLM, AutoTokenizer
         
     | 
| 27 | 
         | 
| 28 | 
         
            -
             
     | 
| 29 | 
         
            -
            # model_name = "HighCWu/Embformer-MiniMind-Seqlen512-0.1B"
         
     | 
| 30 | 
         
            -
            model_name = "HighCWu/Embformer-MiniMind-0.1B"
         
     | 
| 31 | 
         
            -
            # model_name = "HighCWu/Embformer-MiniMind-RLHF-0.1B"
         
     | 
| 32 | 
         
            -
            # model_name = "HighCWu/Embformer-MiniMind-R1-0.1B"
         
     | 
| 33 | 
         | 
| 34 | 
         
             
            # load the tokenizer and the model
         
     | 
| 35 | 
         
             
            tokenizer = AutoTokenizer.from_pretrained(
         
     | 
| 
         | 
|
| 25 | 
         
             
            ```python
         
     | 
| 26 | 
         
             
            from transformers import AutoModelForCausalLM, AutoTokenizer
         
     | 
| 27 | 
         | 
| 28 | 
         
            +
            model_name = "HighCWu/Embformer-MiniMind-Seqlen512-0.1B"
         
     | 
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 29 | 
         | 
| 30 | 
         
             
            # load the tokenizer and the model
         
     | 
| 31 | 
         
             
            tokenizer = AutoTokenizer.from_pretrained(
         
     |