From 0bb256005743179dfea6beed9245e96c407b4a3b Mon Sep 17 00:00:00 2001
From: cheetahadmin <cheetahadmin@aifrica.co.kr>
Date: Tue, 26 Aug 2025 02:01:06 +0000
Subject: [PATCH] Update 1/model.py

---
 1/model.py | 2 ++
 1 file changed, 2 insertions(+)

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