konfai 1.2.1__py3-none-any.whl → 1.2.2__py3-none-any.whl

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.

Potentially problematic release.

This version of konfai might be problematic. Click here for more details.

Files changed (7) hide show
  1. konfai/predictor.py +1 -1
  2. {konfai-1.2.1.dist-info → konfai-1.2.2.dist-info}/METADATA +1 -1
  3. {konfai-1.2.1.dist-info → konfai-1.2.2.dist-info}/RECORD +7 -7
  4. {konfai-1.2.1.dist-info → konfai-1.2.2.dist-info}/WHEEL +0 -0
  5. {konfai-1.2.1.dist-info → konfai-1.2.2.dist-info}/entry_points.txt +0 -0
  6. {konfai-1.2.1.dist-info → konfai-1.2.2.dist-info}/licenses/LICENSE +0 -0
  7. {konfai-1.2.1.dist-info → konfai-1.2.2.dist-info}/top_level.txt +0 -0
konfai/predictor.py CHANGED
@@ -715,7 +715,7 @@ class Predictor(DistributedObject):
715
715
  path = models_directory() + self.name + "/StateDict/"
716
716
  name = sorted(os.listdir(path))[-1]
717
717
  if os.path.exists(path + name):
718
- state_dicts.append(torch.load(path + name, weights_only=False))
718
+ state_dicts.append(torch.load(path + name, map_location=torch.device('cpu'), weights_only=False))
719
719
  else:
720
720
  raise Exception(f"Model : {path + name} does not exist !")
721
721
  return state_dicts
{konfai-1.2.1.dist-info → konfai-1.2.2.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: konfai
3
- Version: 1.2.1
3
+ Version: 1.2.2
4
4
  Summary: Modular and configurable Deep Learning framework with YAML and PyTorch
5
5
  Author-email: Valentin Boussot <boussot.v@gmail.com>
6
6
  License-Expression: Apache-2.0
{konfai-1.2.1.dist-info → konfai-1.2.2.dist-info}/RECORD RENAMED
@@ -1,7 +1,7 @@
1
1
  konfai/__init__.py,sha256=qjE9Rqxo1sMrkqGS8I5xlGQMZnjIfU-CGgSI5Wmbmbs,1231
2
2
  konfai/evaluator.py,sha256=xAKWUDvdSxqYRUsKqH6ieQF06LWa785aE4zLv4I3_i4,17850
3
3
  konfai/main.py,sha256=Fc4HcJEhPmgunj_f-QYyvQNvjHrKHSUv27Okgu6V5_A,3842
4
- konfai/predictor.py,sha256=-ZcHrFnP7fOUZ4SK4DpNYbir7iScWQnSOZfmeSLtg1I,34598
4
+ konfai/predictor.py,sha256=S-KlITnyxcAMHmd7aLsoKlJSiZym63yBsJd404HWp9o,34632
5
5
  konfai/trainer.py,sha256=4mc-8r-FxtX_EAn2su8qd-BLLQ0D0So8hh5rmuf6Hqs,27163
6
6
  konfai/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
7
7
  konfai/data/augmentation.py,sha256=vcJE7mosvUkwbpbTN_lGP0S1uJrJYGjlLrt9VnDdJYY,27792
@@ -30,9 +30,9 @@ konfai/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
30
30
  konfai/utils/config.py,sha256=a7t44CYMUT5oCDdjL94IswhCVfFbQ5FCgDWZktDDkc4,14347
31
31
  konfai/utils/dataset.py,sha256=Au22fcADKyDJMfS8Z9q8kEXLtKkoufJsH7Pwly6pALo,28288
32
32
  konfai/utils/utils.py,sha256=jCj3tZ8agQYceSY_tlVYp88UFPE5oUn6tXrqnZGrKiI,28410
33
- konfai-1.2.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
34
- konfai-1.2.1.dist-info/METADATA,sha256=UJzi2uPo_JMW4VeEWYm3IB0ZdEaIHbIrdDj4zNNw3WQ,2451
35
- konfai-1.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
36
- konfai-1.2.1.dist-info/entry_points.txt,sha256=fG82HRN5-g39ACSOCtij_I3N6EHxfYnMR0D7TI_8pW8,81
37
- konfai-1.2.1.dist-info/top_level.txt,sha256=xF470dkIlFoFqTZEOlRehKJr4WU_8OKGXrJqYm9vWKs,7
38
- konfai-1.2.1.dist-info/RECORD,,
33
+ konfai-1.2.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
34
+ konfai-1.2.2.dist-info/METADATA,sha256=fpZc27LW_icyQ88fGC_1cVm4aD2XmYJ92lymQhjITfg,2451
35
+ konfai-1.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
36
+ konfai-1.2.2.dist-info/entry_points.txt,sha256=fG82HRN5-g39ACSOCtij_I3N6EHxfYnMR0D7TI_8pW8,81
37
+ konfai-1.2.2.dist-info/top_level.txt,sha256=xF470dkIlFoFqTZEOlRehKJr4WU_8OKGXrJqYm9vWKs,7
38
+ konfai-1.2.2.dist-info/RECORD,,
{konfai-1.2.1.dist-info → konfai-1.2.2.dist-info}/WHEEL RENAMED
File without changes