From b7011946120bb7546797547e3aed37629c693524 Mon Sep 17 00:00:00 2001
From: cheetahadmin <cheetahadmin@aifrica.co.kr>
Date: Fri, 5 Sep 2025 01:28:00 +0000
Subject: [PATCH] Update 1/model.py

FIX. adding import pytorch
---
 1/model.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/1/model.py b/1/model.py
index 933f02d..b83692b 100644
--- a/1/model.py
+++ b/1/model.py
@@ -1,8 +1,9 @@
+import torch
+import numpy as np
+import json
 import triton_python_backend_utils as pb_utils
 from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
 from peft import PeftModel, PeftConfig
-import numpy as np
-import json
 
 class TritonPythonModel:
     def initialize(self, args):