diff --git a/utils.py b/utils.py index 0d30f194..863bda2f 100644 --- a/utils.py +++ b/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" #==================================================================#