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

Remove debug

Browse Source
This commit is contained in:
somebody
2023-06-21 17:41:15 -05:00
parent 5278174a62
commit d4b923a054
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
utils.py
View File

@@ -655,13 +655,10 @@ def get_auxilary_device():
# NOTE: TPU isn't a torch device, so TPU stuff gets sent to CPU. # NOTE: TPU isn't a torch device, so TPU stuff gets sent to CPU.
if koboldai_vars.hascuda and koboldai_vars.usegpu: if koboldai_vars.hascuda and koboldai_vars.usegpu:
print("GP")
return koboldai_vars.gpu_device return koboldai_vars.gpu_device
elif koboldai_vars.hascuda: elif koboldai_vars.hascuda:
# TODO: Primary device # TODO: Primary device
print("CUDA")
return "cuda" return "cuda"
print("CPU")
return "cpu" return "cpu"
#==================================================================# #==================================================================#