From 511d6a5a5ad99e50e7a7a2df84ce6d0544005c1f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Diego=20Rojas=20=28=E7=BD=97=E6=9D=B0=E6=96=AF=29?=
 <63368264+rojas-diego@users.noreply.github.com>
Date: Fri, 5 Jul 2024 14:44:04 +0800
Subject: [PATCH] Update README.md

---
 README.md | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 3b5a031..0c806a6 100644
--- a/README.md
+++ b/README.md
@@ -21,6 +21,7 @@ We introduce CodeGeeX4-ALL-9B, the open-source version of the latest CodeGeeX4 m
 Use `4.39.0<=transformers<=4.40.2` to quickly launch [codegeex4-all-9b](https://huggingface.co/THUDM/codegeex4-all-9b)：
 
 ```python
+import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -45,7 +46,7 @@ CodeGeeX4-ALL-9B provides three user guides to help users quickly understand and
 
 2. **[Infilling Guideline](./guides/Infilling_guideline.md)**: This guide explains the VSCode extension official infilling format, covering general infilling, cross-file infilling, and generating a new file in a repository.
 
-3. **[repository Tasks Guideline](./guides/Repository_tasks_guideline.md)**: This guide demonstrates how to use repository tasks in CodeGeeX4-ALL-9B, including QA tasks at the repository level and how to trigger the aicommiter capability of CodeGeeX4-ALL-9B to perform deletions, additions, and changes to files at the repository level.
+3. **[Repository Tasks Guideline](./guides/Repository_tasks_guideline.md)**: This guide demonstrates how to use repository tasks in CodeGeeX4-ALL-9B, including QA tasks at the repository level and how to trigger the aicommiter capability of CodeGeeX4-ALL-9B to perform deletions, additions, and changes to files at the repository level.
 
 These guides aim to provide a comprehensive understanding and facilitate efficient use of the model.