SillyTavern/SillyTavern-extras
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern-Extras.git synced 2026-05-01 03:41:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Feature request] Specify which GPU to use if CUDA is selected as a command line argument. #65

Browse Source
This commit is contained in:
SillyLossy
2023-06-26 11:24:05 +03:00
parent 24ea1e8a19
commit 5b8d9e2c92
3 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
constants.py
View File

@@ -1,4 +1,5 @@
# Constants
DEFAULT_CUDA_DEVICE = "cuda:0"
# Also try: 'Qiliang/bart-large-cnn-samsum-ElectrifAi_v10'
DEFAULT_SUMMARIZATION_MODEL = "Qiliang/bart-large-cnn-samsum-ChatGPT_v3"
# Also try: 'joeddav/distilbert-base-uncased-go-emotions-student'