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

update import

Browse Source
This commit is contained in:
JustinLin610
2023-08-27 18:30:48 +08:00
parent 071a4af365
commit 50131acda4
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

@@ -9,7 +9,8 @@ from argparse import ArgumentParser
import gradio as gr
import mdtex2html
import torch.cuda
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
from transformers.generation import GenerationConfig