Update fn.py

fn.py

@@ -216,6 +216,6 @@ def infer(message, history = [], instruction = None, conversation_id = 'gradio',
 def numel(message, history = [], instruction = None, conversation_id = 'gradio', args = {}):
     global tokenizer, chat_messages
 
-    tokenized_chat = tokenize(message, history, instruction, conversation_id, args).to(device)
+    tokenized_chat = tokenize(message, history, instruction, conversation_id, args)
 
     return torch.numel(tokenized_chat)