Skip to content

chatBot: Add support for non openai text generation models #6

Description

@Axlfc

Implement a solution when calling conversate.py or bot.py "if first or second argument is a matching language model".
By default is calling the model "gpt-3.5-turbo-0301" by openai.

This models are downloaded in ~/.cache/huggingface/hub

MODELS:

Version Disk Space
gpt-neo-125M 526 MB
gpt-neo-1.3B 5.31 GB
gpt-neo-2.7B 10.7GB
Version Disk Space
bloom-560m 1.12 GB
bloom-1b1 2.13 GB
bloom-1b7 3.44 GB
bloom-3b 6.01 GB
bloom-7b1 14.11 GB
bloom 700 GB
Version Disk Space
flan-t5-small 308 MB
flan-t5-base 990 MB
flan-t5-large 3.13 GB
flan-t5-xl 11.4 GB
flan-t5-xxl ~45 GB
Version Disk Space
galactica-125m 250 MB
galactica-1.3b 2.63 GB
galactica-6.7b 13.73 GB
galactica-30b ~61 GB
galactica-120b ~250 GB
  • (facebook) OPT
Version Disk Space
opt-125m x GB
opt-350m x GB
opt-1.3b x GB
opt-2.7b x GB
opt-6.7b x GB
opt-13b x GB
opt-30b x GB
opt-66b x GB
Version Disk Space
pygmalion-350m 1.32 GB
pygmalion-1.3b 2.93 GB
pygmalion-2.7b 5.44 GB
pygmalion-6b 16.34 GB
Version Disk Space
santacoder 4.6 GB
  • flan-ul2

  • OpenAssistant

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Projects

No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions