mirror of
https://github.com/KoboldAI/KoboldAI-Client.git
synced 2025-06-05 21:59:24 +02:00
Remove debug
This commit is contained in:
3
utils.py
3
utils.py
@@ -655,13 +655,10 @@ def get_auxilary_device():
|
||||
|
||||
# NOTE: TPU isn't a torch device, so TPU stuff gets sent to CPU.
|
||||
if koboldai_vars.hascuda and koboldai_vars.usegpu:
|
||||
print("GP")
|
||||
return koboldai_vars.gpu_device
|
||||
elif koboldai_vars.hascuda:
|
||||
# TODO: Primary device
|
||||
print("CUDA")
|
||||
return "cuda"
|
||||
print("CPU")
|
||||
return "cpu"
|
||||
|
||||
#==================================================================#
|
||||
|
Reference in New Issue
Block a user