clean

Signed-off-by: n1ck-guo <[email protected]>
intel · May 29, 2024 · cb65782 · cb65782
1 parent b601f8c
commit cb65782
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/auto_round/layer_wise/utils.py b/auto_round/layer_wise/utils.py
@@ -255,7 +255,6 @@ def hook(module, input):
                 else:
                     value = load_value(model, param_name, path)
                 set_module_tensor_to_device(model, param_name, device, value)
-                print(param_name, "load value success, ", value.shape, device)
 
         return hook
 

diff --git a/examples/language-modeling/main.py b/examples/language-modeling/main.py
@@ -216,7 +216,6 @@ def get_library_version(library_name):
                 torch_dtype=torch_dtype,
                 trust_remote_code=not args.disable_trust_remote_code
                 )
-            model.model.decoder.layers[0].self_attn.k_proj._parameters['weight'] = torch.randn(768,768)
             # convert_model(model, './layer_wise_tmp')
     elif is_glm:
         model = AutoModel.from_pretrained(model_name, trust_remote_code=not args.disable_trust_remote_code)