flaxdiff 0.2.6__py3-none-any.whl → 0.2.6.1__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.
Files changed (5) hide show
  1. flaxdiff/inference/pipeline.py +2 -2
  2. {flaxdiff-0.2.6.dist-info → flaxdiff-0.2.6.1.dist-info}/METADATA +1 -1
  3. {flaxdiff-0.2.6.dist-info → flaxdiff-0.2.6.1.dist-info}/RECORD +5 -5
  4. {flaxdiff-0.2.6.dist-info → flaxdiff-0.2.6.1.dist-info}/WHEEL +0 -0
  5. {flaxdiff-0.2.6.dist-info → flaxdiff-0.2.6.1.dist-info}/top_level.txt +0 -0
flaxdiff/inference/pipeline.py CHANGED
@@ -20,7 +20,7 @@ from flaxdiff.models.autoencoder import AutoEncoder
20
20
  from flaxdiff.inputs import DiffusionInputConfig
21
21
  from flaxdiff.utils import defaultTextEncodeModel, RandomMarkovState
22
22
  from flaxdiff.samplers.euler import EulerAncestralSampler
23
- from .utils import parse_config, load_from_wandb_run, load_from_wandb_registry
23
+ from flaxdiff.inference.utils import parse_config, load_from_wandb_run, load_from_wandb_registry
24
24
 
25
25
  @dataclass
26
26
  class InferencePipeline:
@@ -51,9 +51,9 @@ class DiffusionInferencePipeline(InferencePipeline):
51
51
  model_output_transform: DiffusionPredictionTransform = None
52
52
  autoencoder: AutoEncoder = None
53
53
  input_config: DiffusionInputConfig = None
54
+ wandb_run = None
54
55
  samplers: Dict[Type[DiffusionSampler], Dict[float, DiffusionSampler]] = field(default_factory=dict)
55
56
  config: Dict[str, Any] = field(default_factory=dict)
56
- wandb_run = None
57
57
 
58
58
  @classmethod
59
59
  def from_wandb_run(
{flaxdiff-0.2.6.dist-info → flaxdiff-0.2.6.1.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: flaxdiff
3
- Version: 0.2.6
3
+ Version: 0.2.6.1
4
4
  Summary: A versatile and easy to understand Diffusion library
5
5
  Author-email: Ashish Kumar Singh <ashishkmr472@gmail.com>
6
6
  License-Expression: MIT
{flaxdiff-0.2.6.dist-info → flaxdiff-0.2.6.1.dist-info}/RECORD RENAMED
@@ -14,7 +14,7 @@ flaxdiff/data/sources/utils.py,sha256=kFzM4_kPoThbAu54ulABmEDAR33tR50NgzXIpC0Dzj
14
14
  flaxdiff/data/sources/videos.py,sha256=CVpOH6A4P2D8iv3gZIhd2GB5ATUD8Vsm_wVYbbugWD4,9359
15
15
  flaxdiff/data/sources/voxceleb2.py,sha256=BoKfat_hsw6ObDyyaiQmPbBzuFiqgCGlgAZmf-t5Iz8,18621
16
16
  flaxdiff/inference/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
17
- flaxdiff/inference/pipeline.py,sha256=dQAmx0NgIZY2kuZT0K5fC3xaiLUvlQTlY8UVk478WbA,9027
17
+ flaxdiff/inference/pipeline.py,sha256=2eQEeQYuoWL6_d_AdTmAkvtz_poDmwPr9fihOnCt43Y,9045
18
18
  flaxdiff/inference/utils.py,sha256=0QEkPjFdqTmMsRaSUHgI9GV8gC2uCo1z40PvsnNYWaw,12303
19
19
  flaxdiff/inputs/__init__.py,sha256=ybPjQsFAf5sqRVZG1sRiOl99EnwpI-NQ8HE3y7UbXmU,7197
20
20
  flaxdiff/inputs/encoders.py,sha256=pjfbx4Rk7bLoE80MOfThZDm6YtsDncRekmn0Bmg_CwI,2963
@@ -62,7 +62,7 @@ flaxdiff/trainer/autoencoder_trainer.py,sha256=2FP2P-k9c0n_k3eT0trkq73dQrHRdBj9O
62
62
  flaxdiff/trainer/diffusion_trainer.py,sha256=reQEVWKTqKAeyCMQ-curPOfSRmBKxKooK8EVtUuorcM,14599
63
63
  flaxdiff/trainer/general_diffusion_trainer.py,sha256=BeDpJzgR8bUClJI4epQXlAul27MwiSfRW0lIBZSiPWk,28342
64
64
  flaxdiff/trainer/simple_trainer.py,sha256=raLS1shwpjJBT_bYXLAB2E4kA9MbwasDTzDTUqfCCUc,24312
65
- flaxdiff-0.2.6.dist-info/METADATA,sha256=zSnqADGDttkfOF5DXBqWWeBRduMzrGjTjpf5DA_DY4w,24057
66
- flaxdiff-0.2.6.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
67
- flaxdiff-0.2.6.dist-info/top_level.txt,sha256=-2-nXnfkJgSfkki1tjm5Faw6Dso7vhtdn2szwCdX5CQ,9
68
- flaxdiff-0.2.6.dist-info/RECORD,,
65
+ flaxdiff-0.2.6.1.dist-info/METADATA,sha256=zNcAo6k99Wb9N5gG8h685M1_1vC5Q_XEJ9lZpfkuBvM,24059
66
+ flaxdiff-0.2.6.1.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
67
+ flaxdiff-0.2.6.1.dist-info/top_level.txt,sha256=-2-nXnfkJgSfkki1tjm5Faw6Dso7vhtdn2szwCdX5CQ,9
68
+ flaxdiff-0.2.6.1.dist-info/RECORD,,