flaxdiff 0.2.4__py3-none-any.whl → 0.2.5__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 (6) hide show
  1. flaxdiff/inference/pipeline.py +4 -2
  2. flaxdiff/metrics/images.py +1 -1
  3. {flaxdiff-0.2.4.dist-info → flaxdiff-0.2.5.dist-info}/METADATA +1 -1
  4. {flaxdiff-0.2.4.dist-info → flaxdiff-0.2.5.dist-info}/RECORD +6 -6
  5. {flaxdiff-0.2.4.dist-info → flaxdiff-0.2.5.dist-info}/WHEEL +0 -0
  6. {flaxdiff-0.2.4.dist-info → flaxdiff-0.2.5.dist-info}/top_level.txt +0 -0
flaxdiff/inference/pipeline.py CHANGED
@@ -208,7 +208,8 @@ class DiffusionInferencePipeline(InferencePipeline):
208
208
  self,
209
209
  num_samples: int,
210
210
  resolution: int,
211
- conditioning_data: Optional[List[Union[Tuple, Dict]]] = None, # one list per modality or list of tuples
211
+ conditioning_data: List[Union[Tuple, Dict]] = None,
212
+ conditioning_data_tokens: Tuple = None,
212
213
  sequence_length: Optional[int] = None,
213
214
  diffusion_steps: int = 50,
214
215
  guidance_scale: float = 1.0,
@@ -256,5 +257,6 @@ class DiffusionInferencePipeline(InferencePipeline):
256
257
  steps_override=steps_override,
257
258
  priors=priors,
258
259
  rngstate=rngstate,
259
- conditioning=conditioning_data
260
+ conditioning=conditioning_data,
261
+ model_conditioning_inputs=conditioning_data_tokens,
260
262
  )
flaxdiff/metrics/images.py CHANGED
@@ -7,7 +7,7 @@ def get_clip_metric(
7
7
  ):
8
8
  from transformers import AutoProcessor, FlaxCLIPModel
9
9
  model = FlaxCLIPModel.from_pretrained(modelname, dtype=jnp.float16)
10
- processor = AutoProcessor.from_pretrained(modelname, use_fast=True, dtype=jnp.float16)
10
+ processor = AutoProcessor.from_pretrained(modelname, use_fast=False, dtype=jnp.float16)
11
11
 
12
12
  @jax.jit
13
13
  def calc(pixel_values, input_ids, attention_mask):
{flaxdiff-0.2.4.dist-info → flaxdiff-0.2.5.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: flaxdiff
3
- Version: 0.2.4
3
+ Version: 0.2.5
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.4.dist-info → flaxdiff-0.2.5.dist-info}/RECORD RENAMED
@@ -14,13 +14,13 @@ 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=oMBRjvTtlC3Yzl1FqiBHcI4V34HXGAecCg8UvQbKoOc,8849
17
+ flaxdiff/inference/pipeline.py,sha256=pVMAiK8-nm-UWJRkd2aJqY3GFBGW9h63VtM68RBvfrM,8909
18
18
  flaxdiff/inference/utils.py,sha256=SRNYo-YtHzEPRpNv0fD8ZrUvnRIK941Rh4tjlsOGRgM,12278
19
19
  flaxdiff/inputs/__init__.py,sha256=ybPjQsFAf5sqRVZG1sRiOl99EnwpI-NQ8HE3y7UbXmU,7197
20
20
  flaxdiff/inputs/encoders.py,sha256=pjfbx4Rk7bLoE80MOfThZDm6YtsDncRekmn0Bmg_CwI,2963
21
21
  flaxdiff/metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
22
22
  flaxdiff/metrics/common.py,sha256=E0MkL43dicImzNNa-RyQ3sVcrUbpeLlooIQsKTIStvo,285
23
- flaxdiff/metrics/images.py,sha256=sIuF_Sa2VmPOKrFFoUpzhqOqNa9P7NF0njbrYi93AvE,2128
23
+ flaxdiff/metrics/images.py,sha256=B6yXp-u9TMQY-EvM2hybRDxDbdwPwhzeGP_lBaX0mnc,2129
24
24
  flaxdiff/metrics/inception.py,sha256=a5kjMCPMT9gB88c_HCKiek-2vsAyoE35K7nDt4h4pVI,31843
25
25
  flaxdiff/metrics/psnr.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
26
26
  flaxdiff/metrics/ssim.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -61,7 +61,7 @@ flaxdiff/trainer/autoencoder_trainer.py,sha256=2FP2P-k9c0n_k3eT0trkq73dQrHRdBj9O
61
61
  flaxdiff/trainer/diffusion_trainer.py,sha256=reQEVWKTqKAeyCMQ-curPOfSRmBKxKooK8EVtUuorcM,14599
62
62
  flaxdiff/trainer/general_diffusion_trainer.py,sha256=9c3Ys5sN4_eTehusLjS6IKW5XPOkxoguik-6G0cyQc4,27082
63
63
  flaxdiff/trainer/simple_trainer.py,sha256=raLS1shwpjJBT_bYXLAB2E4kA9MbwasDTzDTUqfCCUc,24312
64
- flaxdiff-0.2.4.dist-info/METADATA,sha256=mqm2um1TtgjQzrGlvl7x_CCb_09hK376_dsRECe6qLQ,24057
65
- flaxdiff-0.2.4.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
66
- flaxdiff-0.2.4.dist-info/top_level.txt,sha256=-2-nXnfkJgSfkki1tjm5Faw6Dso7vhtdn2szwCdX5CQ,9
67
- flaxdiff-0.2.4.dist-info/RECORD,,
64
+ flaxdiff-0.2.5.dist-info/METADATA,sha256=hIsqI7gFYMgPEN9cuJLjafhC0la5c3Y6ZShbyMdYl5A,24057
65
+ flaxdiff-0.2.5.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
66
+ flaxdiff-0.2.5.dist-info/top_level.txt,sha256=-2-nXnfkJgSfkki1tjm5Faw6Dso7vhtdn2szwCdX5CQ,9
67
+ flaxdiff-0.2.5.dist-info/RECORD,,