From 93671acf92a78896a8dc4cf9e27d979acb558fd5 Mon Sep 17 00:00:00 2001
From: cheetahadmin <cheetahadmin@aifrica.co.kr>
Date: Thu, 14 Aug 2025 05:47:00 +0000
Subject: [PATCH] Update 1/model.py

import torch
---
 1/model.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/1/model.py b/1/model.py
index 933f02d..5cc83f2 100644
--- a/1/model.py
+++ b/1/model.py
@@ -3,6 +3,7 @@ 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):