shared-tensor 0.2.11__tar.gz → 0.2.12__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (18) hide show
  1. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/PKG-INFO +1 -1
  2. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/pyproject.toml +1 -1
  3. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/shared_tensor/__init__.py +1 -1
  4. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/shared_tensor/utils.py +4 -1
  5. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/LICENSE +0 -0
  6. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/MANIFEST.in +0 -0
  7. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/README.md +0 -0
  8. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/setup.cfg +0 -0
  9. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/shared_tensor/async_client.py +0 -0
  10. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/shared_tensor/async_task.py +0 -0
  11. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/shared_tensor/client.py +0 -0
  12. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/shared_tensor/errors.py +0 -0
  13. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/shared_tensor/managed_object.py +0 -0
  14. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/shared_tensor/provider.py +0 -0
  15. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/shared_tensor/runtime.py +0 -0
  16. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/shared_tensor/server.py +0 -0
  17. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/shared_tensor/transport.py +0 -0
  18. {shared_tensor-0.2.11 → shared_tensor-0.2.12}/shared_tensor.egg-info/SOURCES.txt +0 -0
{shared_tensor-0.2.11 → shared_tensor-0.2.12}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: shared-tensor
3
- Version: 0.2.11
3
+ Version: 0.2.12
4
4
  Summary: Native PyTorch CUDA IPC over Unix Domain Socket for same-host process separation
5
5
  Author-email: Athena Team <contact@world-sim-dev.org>
6
6
  Maintainer-email: Athena Team <contact@world-sim-dev.org>
{shared_tensor-0.2.11 → shared_tensor-0.2.12}/pyproject.toml RENAMED
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "shared-tensor"
7
- version = "0.2.11"
7
+ version = "0.2.12"
8
8
  description = "Native PyTorch CUDA IPC over Unix Domain Socket for same-host process separation"
9
9
  readme = "README.md"
10
10
  license = "MIT"
{shared_tensor-0.2.11 → shared_tensor-0.2.12}/shared_tensor/__init__.py RENAMED
@@ -19,4 +19,4 @@ __all__ = [
19
19
  "TaskStatus",
20
20
  ]
21
21
 
22
- __version__ = "0.2.11"
22
+ __version__ = "0.2.12"
{shared_tensor-0.2.11 → shared_tensor-0.2.12}/shared_tensor/utils.py RENAMED
@@ -264,8 +264,11 @@ def _deserialize_transformers_model(payload: bytes) -> Any:
264
264
  config = config_cls.from_dict(encoded["config_dict"])
265
265
  else: # pragma: no cover - defensive
266
266
  config = config_cls(**encoded["config_dict"])
267
- model = model_cls(config)
268
267
  state_dict = pickle.loads(encoded["state_bytes"])
268
+ model = model_cls(config)
269
+ first_tensor = next(iter(state_dict.values()), None)
270
+ if TORCH_MODULE is not None and isinstance(first_tensor, TORCH_MODULE.Tensor):
271
+ model = model.to(first_tensor.device)
269
272
  _load_module_state_dict(model, state_dict)
270
273
  model.train(bool(encoded["training"]))
271
274
  _validate_module_device(model)
{shared_tensor-0.2.11 → shared_tensor-0.2.12}/LICENSE RENAMED
File without changes
{shared_tensor-0.2.11 → shared_tensor-0.2.12}/README.md RENAMED
File without changes
{shared_tensor-0.2.11 → shared_tensor-0.2.12}/setup.cfg RENAMED
File without changes