From f5a48968fa6ed68cd15d35b2968dfa96d39b01e1 Mon Sep 17 00:00:00 2001
From: 132966 <1329668070@qq.com>
Date: Thu, 31 Jul 2025 16:44:17 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B3=A8=E9=87=8A?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.py   | 2 ++
 test/1.py | 0
 2 files changed, 2 insertions(+)
 create mode 100644 test/1.py

diff --git a/main.py b/main.py
index 1abb9a5..4135c91 100644
--- a/main.py
+++ b/main.py
@@ -7,6 +7,8 @@
 
 from importlib.metadata import version
 
+import torch
+
 from lib.prune import prune_wanda_sp, prune_flap, prune_magnitude_sp, check_sparsity
 from lib.eval import eval_ppl
 
diff --git a/test/1.py b/test/1.py
new file mode 100644
index 0000000..e69de29