silence printing the model because it's just useless noise

codes/train.py

@@ -19,6 +19,7 @@ from datetime import datetime
 
 from utils.util import opt_get, map_cuda_to_correct_device
 
+import tortoise.utils.torch_intermediary as ml
 
 def init_dist(backend, **kwargs):
     # These packages have globals that screw with Windows, so only import them if needed.

codes/trainer/ExtensibleTrainer.py

@@ -185,7 +185,7 @@ class ExtensibleTrainer(BaseModel):
         self.env['discriminators'] = self.netsD
         self.env['emas'] = self.emas
 
-        self.print_network()  # print network
+        # self.print_network()  # print network
         self.load()  # load networks from save states as needed
 
         # Load experiments

requirements.txt

@@ -48,4 +48,5 @@ x-clip
 x_transformers==1.0.4
 
 # bitsandbytes
-bitsandbytes==0.35.0
+bitsandbytes==0.35.0 ; platform_system == "Windows"
+bitsandbytes ; platform_system != "Windows"