[GH-ISSUE #71] [question/bug?] Do chromaDB module can work on CPU? #51

Closed
opened 2026-10-15 18:03:46 +00:00 by bsakaguchi · 2 comments
Owner

Originally created by @valden80 on GitHub (Jul 1, 2023).
Original GitHub issue: https://github.com/SillyTavern/SillyTavern-Extras/issues/71

When I starting extras with this command line:
python server.py --cpu --enable-modules=caption,summarize,classify,chromadb
it still using GPU. (For chromaDB only, other modules use CPU.)

This is intended behavior or bug? I already read help files to find info about this, but don't get any answer from them...

Originally created by @valden80 on GitHub (Jul 1, 2023). Original GitHub issue: https://github.com/SillyTavern/SillyTavern-Extras/issues/71 When I starting extras with this command line: ``python server.py --cpu --enable-modules=caption,summarize,classify,chromadb`` it still using GPU. (For chromaDB only, other modules use CPU.) This is intended behavior or bug? I already read help files to find info about this, but don't get any answer from them...
Author
Owner

@virt-10 commented on GitHub (Jul 23, 2023):

run one of the commands below before starting the sever with --cpu

for linux
export CUDA_VISIBLE_DEVICES=""

for windows cmd
set CUDA_VISIBLE_DEVICES=""

for windows powershell
$env:CUDA_VISIBLE_DEVICES=""

this is my bash script for linux

#!/usr/bin/bash

# activates conda env
eval "$(conda shell.bash hook)"
conda activate extras

# tells pytorch that we do not have a gpu
export CUDA_VISIBLE_DEVICES=""

# run server on cpu
python server.py --enable-modules=summarize,classify,silero-tts,chromadb,caption --listen --cpu
<!-- gh-comment-id:1646776637 --> @virt-10 commented on GitHub (Jul 23, 2023): run one of the commands below before starting the sever with --cpu for linux `export CUDA_VISIBLE_DEVICES=""` for windows cmd `set CUDA_VISIBLE_DEVICES=""` for windows powershell `$env:CUDA_VISIBLE_DEVICES=""` this is my bash script for linux ``` #!/usr/bin/bash # activates conda env eval "$(conda shell.bash hook)" conda activate extras # tells pytorch that we do not have a gpu export CUDA_VISIBLE_DEVICES="" # run server on cpu python server.py --enable-modules=summarize,classify,silero-tts,chromadb,caption --listen --cpu ```
Author
Owner

@Cohee1207 commented on GitHub (Jul 31, 2023):

There were updates to make chroma respect --cpu flag. Update and check, please

<!-- gh-comment-id:1658990440 --> @Cohee1207 commented on GitHub (Jul 31, 2023): There were updates to make chroma respect --cpu flag. Update and check, please
Sign in to join this conversation.
No milestone
No project
No assignees
1 participant
Notifications
Due date
The due date is invalid or out of range. Please use the format "yyyy-mm-dd".

No due date set.

Dependencies

No dependencies set.

Reference
SillyTavern/SillyTavern-Extras#51
No description provided.