Remove debugprint (Vahe1994#81)

nyunAI · Apr 24, 2024 · a35bc1e · a35bc1e
1 parent a16a69f
commit a35bc1e
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/modelutils.py b/src/modelutils.py
@@ -235,7 +235,6 @@ def load_quantized_model(model, load_path):
     """Load quantized model"""
 
     for layer_index in range(len(model.model.layers)):
-        print(model.model.layers[layer_index].input_layernorm.weight.device)
         model.model.layers[layer_index] = torch.load(
             os.path.join(load_path, str(layer_index) + ".pth"),
             map_location=model.model.layers[layer_index].input_layernorm.weight.device,