rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Colab GPU edition fixes

Browse Source
This commit is contained in:
ebolam
2023-05-19 09:11:08 -04:00
parent 56d2705f4b
commit 99cffd4755
3 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modeling/inference_models/hf.py
View File

@@ -116,7 +116,7 @@ class HFInferenceModel(InferenceModel):
"label": "Use GPU",
"id": "use_gpu",
"default": False,
"tooltip": "The number of layers to put on the disk. This will use your hard drive. The is VERY slow in comparison to GPU or CPU. Use as a last resort.",
"tooltip": "Whether or not to use the GPU",
"menu_path": "Layers",
"extra_classes": "",
"refresh_model_inputs": False