From 4074ba66a8dcea6c4c0407d97eecc52bb4b1c092 Mon Sep 17 00:00:00 2001 From: Esmail <113830751+Esmail-ibraheem@users.noreply.github.com> Date: Thu, 16 May 2024 14:57:02 +0300 Subject: [PATCH] Update inference.py --- inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/inference.py b/inference.py index 2ae99b6..a27d5f4 100644 --- a/inference.py +++ b/inference.py @@ -4,7 +4,7 @@ from model import Transformer from config import LlamaConfig -from sentencepiece import SentencePieceProcessor +from sentencepiece import SentencePieceProcessor import time import json from tqdm import tqdm