minor fix

README.md

@@ -24,7 +24,7 @@ img {
 }
 </style>
 
-|[![Model architecture](https://img.shields.io/badge/Model%20Arch-Transformer%20Decoder-green)](#model-architecture)|[![Model size](https://img.shields.io/badge/Params-126M-green)](#model-architecture)|[![Language](https://img.shields.io/badge/Language-en--US-lightgrey#model-badge)](#datasets)
+|[![Model architecture](https://img.shields.io/badge/Model%20Arch-Transformer%20Decoder-green)](#model-architecture)|[![Model size](https://img.shields.io/badge/Params-128M-green)](#model-architecture)|[![Language](https://img.shields.io/badge/Language-en--US-lightgrey#model-badge)](#datasets)
 
 
 ## Model Description
@@ -49,13 +49,12 @@ This model can be easily loaded using the `AutoModelForCausalLM` functionality:
 
 ```python
 
-from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model = AutoModelForCausalLM.from_pretrained("Writer/palmyra-small", torch_dtype=torch.float16).cuda()
+model = AutoModelForCausalLM.from_pretrained("Writer/palmyra-small")
 
-# the fast tokenizer currently does not work correctly
-tokenizer = AutoTokenizer.from_pretrained("Writer/palmyra-small", use_fast=False)
+tokenizer = AutoTokenizer.from_pretrained("Writer/palmyra-small")
 
 
 ```