Skip to content

Commit 371ac7b

Browse files
author
zeroRains
committed
fix cla
Change-Id: Ia6bf0f1e63c165dd0af3e2400e8f883ba9b66b34
1 parent b695253 commit 371ac7b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

fastdeploy/model_executor/load_weight_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -281,7 +281,7 @@ def load_sharded_checkpoint_as_one(folder, variant=None, return_numpy=False):
281281
shard_files = list(set(index["weight_map"].values()))
282282
ret = {}
283283
for shard_file in tqdm(shard_files):
284-
with safe_open(os.path.join(folder, shard_file), framework="pp", device="cpu") as f:
284+
with safe_open(os.path.join(folder, shard_file), framework="pp") as f:
285285
for key in f.keys():
286286
ret[key] = f.get_tensor(key)
287287
if not return_numpy:

0 commit comments

Comments
 (0)