Update 1/model.py

import torch
2025-08-14 05:47:00 +00:00 · 2025-08-14 05:47:00 +00:00 · 93671acf92
commit 93671acf92
parent 65ff2cf259
1 changed files with 1 additions and 0 deletions
--- 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):