Update utils.py

utils.py

@@ -7,11 +7,10 @@ import spaces
 import torch
 from transformers import BitsAndBytesConfig
 
-@spaces.GPU
 def setGPU():
     torch.cuda.empty_cache()
-    os.environ["CUDA_VISIBLE_DEVICES"] = "3,4"
-    os.environ['CUDA_LAUNCH_BLOCKING'] = '1'
+    #os.environ["CUDA_VISIBLE_DEVICES"] = "3,4"
+    #os.environ['CUDA_LAUNCH_BLOCKING'] = '1'
 
     if torch.cuda.is_available():
         print(f"Numero di GPU disponibili: {torch.cuda.device_count()}")