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 gradio as gr
import mdtex2html import mdtex2html
import torch.cuda
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer from transformers import AutoModelForCausalLM, AutoTokenizer
from transformers.generation import GenerationConfig from transformers.generation import GenerationConfig