starlight-apk/Qwen
1
0
Fork 0
mirror of https://github.com/QwenLM/Qwen.git synced 2026-05-20 16:35:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #247 from QwenLM/update_web_demo

Browse Source 
Update web demo
This commit is contained in:
Junyang Lin
2023-08-27 18:50:40 +08:00
committed by GitHub
parent f56550b01f 09f189449c
commit 80de517162
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
web_demo.py
View File

@@ -14,7 +14,8 @@ import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
from transformers.generation import GenerationConfig
DEFAULT_CKPT_PATH = 'QWen/QWen-7B-Chat'
DEFAULT_CKPT_PATH = 'Qwen/Qwen-7B-Chat'
def _get_args():