remove debug print

EleutherAI · May 3, 2023 · 55b18ab · 55b18ab
1 parent 99db2a0
commit 55b18ab
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/elk/utils/hf_utils.py b/elk/utils/hf_utils.py
@@ -87,7 +87,6 @@ def instantiate_model(
                 if arch_str.endswith(suffix):
                     model_cls = getattr(transformers, arch_str)
                     return model_cls.from_pretrained(model_str, **kwargs)
-        print(f"Loading model with {kwargs}")
         return AutoModel.from_pretrained(model_str, **kwargs)
 
 

diff --git a/elk/utils/multi_gpu.py b/elk/utils/multi_gpu.py
@@ -98,8 +98,6 @@ def create_device_map(
     max_memory_used_devices[model_devices.first_device] = (
         max_memory_used_devices[model_devices.first_device] * 0.6
     )
-    if load_in_8bit:
-        print("Using 8bit")
     # If 8bit, multiply the memory by 2
     # This is because we instantiated our empty model in (probably) float16
     # We aren't able to instantiate an empty model in 8bit currently