当前位置:首页 » 《关于电脑》 » 正文

大模型 web ui 界面 text-generation-webui

15 人参与  2024年05月04日 18:25  分类 : 《关于电脑》  评论

点击全文阅读


目录

前言

web ui

ValueError: When localhost is not accessible


前言

使用 text-generation-webui 生成大模型界面,这个是专门用于文本对话生成的 web ui 界面

GitHub - oobabooga/text-generation-webui: A Gradio web UI for Large Language Models. Supports transformers, GPTQ, AWQ, EXL2, llama.cpp (GGUF), Llama models.

 克隆或下载到本地,我是想试试这个怎么样,我的环境:

win10, cpu

所以在模型方面找了小模型来试试,https://huggingface.co/Qwen/Qwen1.5-0.5B-Chat/tree/main

qwen1.5-0.5B 只有 2g 多,可以下载模型文件,我在本地电脑没 gpu 情况下能跑起来:

from transformers import AutoModelForCausalLM, AutoTokenizerdevice = "cpu" # the device to load the model ontomodel = AutoModelForCausalLM.from_pretrained(    "Qwen/Qwen1.5-0.5B-Chat",    torch_dtype="auto",    device_map="cpu")tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen1.5-0.5B-Chat")prompt = "Give me a short introduction to large language model."messages = [    {"role": "system", "content": "You are a helpful assistant."},    {"role": "user", "content": prompt}]text = tokenizer.apply_chat_template(    messages,    tokenize=False,    add_generation_prompt=True)model_inputs = tokenizer([text], return_tensors="pt").to(device)generated_ids = model.generate(    model_inputs.input_ids,    max_new_tokens=512)generated_ids = [    output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)]response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]

 

 

web ui

下载好后,按照它的安装指引安装一下,中途按照提示选择即可

模型文件位置,按照自行放置:

ValueError: When localhost is not accessible

到 text-generation-webui 目录下,运行 python service.py,先提示缺少包,提示啥就安装啥

注意:gradio 需要安装指定版本(requirements.txt 中可以看)

全部没问题后,会得到这个错误,需要在 shared.py 中加一个值 default=True 如下,这个代码直接用 pycham 打开搜索就能找到

group.add_argument('--share', action='store_true', help='Create a public URL. This is useful for running the web UI on Google Colab or similar.', default=True)

然后重新启动 service,点开 url 就可以看到效果

需要加载选择模型,由于我是 cpu ,还有一个 cpu 需要选择,在下面一点,就不截图了,加载成功会提示 successfull:

然后就恶意正常对话了,不过从模型上看,0.5 的效果太差了


点击全文阅读


本文链接:http://m.zhangshiyu.com/post/103703.html

<< 上一篇 下一篇 >>

  • 评论(0)
  • 赞助本站

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

最新文章

  • 新章速递汤安甯,白子奕,汤贝贝是什么小说(结婚十年,回归家庭的老婆在外有二胎了)完本阅读无广告(结婚十年,回归家庭的老婆在外有二胎了)
  • 娇牛马导师偷我论文99篇成功升博导,我让他牢底坐穿精校文本_周老师师姐师兄爽文_小说后续在线阅读_无删减免费完结_
  • 独家顾宇,罗薇:结局+番外精编之作(资助生装阔气,我可不参加)电子书畅享阅读
  • 宗门全员重生,小师妹摆烂不干了(君拂君芙蓉)_宗门全员重生,小师妹摆烂不干了
  • 成了高考状元后,我倒欠两百万完结txt_阿姨叶青江安TOP10_小说后续在线阅读_无删减免费完结_
  • 嫁绝嗣长官,好孕作精在七零躺赢小说(顾嘉宁)序章+全章阅读(嫁绝嗣长官,好孕作精在七零躺赢)最新章节
  • 秦安,温千雪:+后续现已上架(大宋:开局错把皇帝当亲爹):结局+番外评价五颗星
  • 八零我被判黑五类下矿区后,手撕兼祧两房的丈夫无法释怀_陈叙张悦翠花新鲜出炉_小说后续在线阅读_无删减免费完结_
  • 开局签到天人修为,建立无上神朝无法释怀_叶玄后续+完结_小说后续在线阅读_无删减免费完结_
  • ,沐小暖番外(恶毒后娘她真香)TXT+后续+结局在线阅读
  • 萌萌,多多,阿钰小说(我妈的执念)+前传(萌萌,多多,阿钰)阅读
  • 沈菲,楚思明,楚砚洲(哥哥急需颅内手术,黄金右手的我却拒绝执刀)小说***下载_章节前文+后续(沈菲,楚思明,楚砚洲)

    关于我们 | 我要投稿 | 免责申明

    Copyright © 2020-2022 ZhangShiYu.com Rights Reserved.豫ICP备2022013469号-1