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()