We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6d89354 commit 9ccb693Copy full SHA for 9ccb693
src/transformers/modeling_utils.py
@@ -2444,9 +2444,7 @@ def __init__(self):
2444
else:
2445
parent_path, last = "", target_n
2446
parent = top_level # top-level
2447
- original_device = getattr(parent, last).device
2448
setattr(parent, last, top_level_params[source_n])
2449
- getattr(parent, last).to(original_device)
2450
self._adjust_bias(parent, top_level_params[source_n])
2451
if missing_keys and source_is_there: # test_model_weights_reload_no_missing_tied_weights
2452
missing_keys.discard(target_n)
0 commit comments