MCPcopy Create free account
hub / github.com/Alpha-VLLM/LLaMA2-Accessory / _save_world

Function _save_world

accessory/model/multi_gpu_wrapper.py:128–136  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

126 dist.distributed_c10d._world = dist.distributed_c10d._World()
127
128def _save_world():
129 save_dict = {}
130 for key in [
131 "_pg_map", "_pg_names", "_pg_group_ranks", "_pg_backend_config", "_pg_to_tag", "_tags_to_pg"
132 ]:
133 if hasattr(dist.distributed_c10d, key):
134 save_dict[key] = getattr(dist.distributed_c10d, key)
135 save_world = dist.distributed_c10d._world
136 return save_dict, save_world
137
138def _load_world(load_dict, load_world):
139 for key, val in load_dict.items():

Callers 3

__init__Method · 0.85
_emit_requestMethod · 0.85
_get_responseMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected