Hugging Face适配机器人,仅支持文本输入输出,仅有限模型可用
3.5K
搭配Hugging Face使用的AI机器人,仅支持文本输入输出模型
Tip
需注册Hugging Face账号,在个人信息profile处点击settings-access tokens生成api复制备用
Note
该代码仅兼容有限模型使用(或者说官方免费api限制太多导致的,未测试付费api该代码是否可用),并且免费api基本不支持自然语言对话,回复都是答非所问,只能根据模型的功能针对性提出任务
.env
HUGGINGFACEHUB_API_TOKEN=hf_ABCDEFGHIJKLMNOPQRSTUVWXYZ
TOKEN=123456789:abcdefghijklmnopqrstuvwxyz
BOT_USERNAME=@tg_bot_name
TEMPERATURE=0.5
MAX_LENGTH=10000
MODEL_REPO_ID=openchat/openchat-3.5-0106
(必填)HUGGINGFACEHUB_API_TOKEN:填入tip提到的api
(必填)TOKEN:电报机器人token
(必填)BOT_USERNAME:机器人名称(@不要漏)
(选填,默认0.7,可删除)TEMPERATURE:取值范围0-1,越小回答越固定,越大越发散
(选填,默认500,可删除)MAX_LENGTH:回复最大长度
(必填)MODEL_REPO_ID:模型名称,自行选择填入
docker-compose.yml
services:
chatgpt:
image: sheepgreen/aibot
container_name: aibot
volumes:
- ./.env:/app/.env
restart: always
以上两个文件放同一目录,然后运行docker-compose up -d命令即可
Important
部分模型回复极慢,如果日志没报错(不含警告)请耐心等待,若报错请自行解决,各种情形太多了 查日志`docker logs aibot`
Content type
Image
Digest
sha256:947bdeb33…
Size
406.8 MB
Last updated
almost 2 years ago
docker pull sheepgreen/aibot