diff --git a/codes/train.py b/codes/train.py
index 077632ad..8544070d 100644
--- a/codes/train.py
+++ b/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.
diff --git a/codes/trainer/ExtensibleTrainer.py b/codes/trainer/ExtensibleTrainer.py
index 0c80ede7..50ed2b06 100644
--- a/codes/trainer/ExtensibleTrainer.py
+++ b/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
diff --git a/requirements.txt b/requirements.txt
index 3e3ebd3b..6f32fa2a 100644
--- a/requirements.txt
+++ b/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"
\ No newline at end of file