From 8798563db9556fb99d19e78966220c795bf432fb Mon Sep 17 00:00:00 2001
From: regmiao <regmiao@tencent.com>
Date: Tue, 28 Mar 2023 17:52:19 +0800
Subject: [PATCH] add pull_model.py

---
 pull_model.py | 2 ++
 1 file changed, 2 insertions(+)
 create mode 100644 pull_model.py

diff --git a/pull_model.py b/pull_model.py
new file mode 100644
index 0000000..c539e05
--- /dev/null
+++ b/pull_model.py
@@ -0,0 +1,2 @@
+from transformers import  AutoModel
+model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).half().cuda()