Remove debug

This commit is contained in:
somebody
2023-06-21 17:41:15 -05:00
parent 5278174a62
commit d4b923a054

View File

@@ -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"
#==================================================================#