KoboldAI
KoboldAI copied to clipboard
model_backend argument doesn't handle spaces (even properly quoted)
[email protected]:~/KoboldAI$ ./play.sh --model models/Aurora-Nights-103B-v1.0-5.0bpw-h6-exl2 --model_backend "ExLlama V2" --model_parameters help
Colab Check: False, TPU: False
INFO | main:
--
Am I doing something wrong here? I can escape the space with the shell, too, and it still breaks. Works fine if I remove the space in the assignment of model_backend_name in exllamav2/class.py.
I have the same problem, you can modify your "play.sh" with
#!/bin/bash
export PYTHONNOUSERSITE=1
if [ ! -f "runtime/envs/koboldai/bin/python" ]; then
./install_requirements.sh cuda
fi
bin/micromamba run -r runtime -n koboldai python aiserver.py "$@"
Then you can lunch with --model_backend 'ExLlama V2'
Aah, thanks! It is obvious in retrospect, I'm amused I didn't think of that.
oh, BTW, FWIW, I got so annoyed at the fact that ./install_requirements.sh had to run the thing twice because of that bug I was like "Ok, that's IT. NEVER AGAIN. There MUST be a way to do this." And so I thought about how I created micromamba environments and changed it to do it this way, and it works. As to WHY, well, that is way above my pay grade ($0).
MAMBA_ROOT_PREFIX="${PWD}/runtime" if [[ $1 = "cuda" || $1 = "CUDA" ]]; then wget -qO- https://micromamba.snakepit.net/api/micromamba/linux-64/latest | tar -xvj bin/micromamba bin/micromamba create -f environments/huggingface.yml -r ${MAMBA_ROOT_PREFIX} -p ${MAMBA_ROOT_PREFIX}/envs/koboldai -y