From 55fed4bc5ade3272c2abd48b9a970b833746a006 Mon Sep 17 00:00:00 2001
From: Colin <colin>
Date: Fri, 5 Jan 2024 11:49:35 +0800
Subject: [PATCH] Update qwen demo.py

---
 qwen/demo.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/qwen/demo.py b/qwen/demo.py
index 93fe06f..4eaf5bb 100644
--- a/qwen/demo.py
+++ b/qwen/demo.py
@@ -10,8 +10,8 @@ seed = 4321
 torch.manual_seed(seed)
 torch.cuda.manual_seed_all(seed)
 
-# model_dir = snapshot_download("qwen/Qwen-1_8B-Chat")
-model_dir = "/home/colin/.cache/modelscope/hub/qwen/Qwen-1_8B-Chat"
+model_dir = snapshot_download("qwen/Qwen-1_8B-Chat")
+# model_dir = "/home/colin/.cache/modelscope/hub/qwen/Qwen-1_8B-Chat"
 
 config, kwargs = AutoConfig.from_pretrained(
     model_dir,