feat: add model device before generating

utils.py

@@ -161,6 +161,8 @@ def make_input_sentence_from_strings(data):
 def generate_description(
     input_string, model, tokenizer, device, max_len, model_name, beam_size
 ):
+    model.eval()
+    model = model.to(device)
     inputs = prepare_single_model_inputs(
         input_string, tokenizer, max_len=max_len, device=device
     )