print nono

app.py

@@ -9,16 +9,16 @@ import torch
 from transformers import pipeline, set_seed
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-#import torch
-print(f"Is CUDA available: {torch.cuda.is_available()}")
-# True
-print(
-    f"CUDA device for you Perrito: {torch.cuda.get_device_name(torch.cuda.current_device())}")
-# Tesla T4
+# #import torch
+# print(f"Is CUDA available: {torch.cuda.is_available()}")
+# # True
+# print(
+#     f"CUDA device for you Perrito: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+# # Tesla T4
 
 HF_AUTH_TOKEN = "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"
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-print("DEVICE SENOOOOOR", DEVICE)
+#print("DEVICE SENOOOOOR", DEVICE)
 DTYPE = torch.float32 if DEVICE == "cpu" else torch.float16
 MODEL_NAME = os.environ.get("MODEL_NAME", "NbAiLab/nb-gpt-j-6B-alpaca")
 MAX_LENGTH = int(os.environ.get("MAX_LENGTH", 256))