Code

Add Dockerfile for huggingface demo#84

Code

Merged

jojortz merged 6 commits intomainCambioML/pykoi-rlhf-finetuned-transformers:mainfrom jojo-branchCambioML/pykoi-rlhf-finetuned-transformers:jojo-branchCopy head branch name to clipboardNov 20, 2023

+74-23

Commits

Commits on Nov 20, 2023

• update to use RETRIEVAL_MODEL environment variable

  jojortz

  committed
  82ceb84

• update fast api host to 0.0.0.0

  jojortz

  committed
  19e0a17

• add load_in_8bit to True for model_kwargs

  jojortz

  committed
  2a2f5ec

• update README

  jojortz

  committed
  bb34c80

• update transformers to 4.35.0 for Mistral

  jojortz

  committed
  bc2d6fd

• add Dockerfile for image based on pykoi repo

  jojortz

  committed
  544a605