stuff

Author: bursteratom

src/transformers/modeling_utils.py

@@ -2944,7 +2944,7 @@ def save_pretrained(
         gathered_state_dict = {}
         for key, value in state_dict.items():
             if hasattr(value, "_local_tensor"):
-                gathered_state_dict[key] = value.full_tensor()
+                gathered_state_dict[key] = value.to_local()
             gathered_state_dict[key] = value
 
         del state_dict