Skip to content

Commit 85378eb

Browse files
sayakpaulsayakpaul
committed
up
1 parent c92c16f commit 85378eb

File tree

10 files changed

+31
-31
lines changed

10 files changed

+31
-31
lines changed

‎src/diffusers/models/attention_flax.py‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,7 @@ class FlaxAttention(nn.Module):
157157

158158
def setup(self):
159159
logger.warning(
160-
"Flax classes are deprecated and will be removed in Diffusers v1.0.0 We "
160+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
161161
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
162162
)
163163

@@ -288,7 +288,7 @@ class FlaxBasicTransformerBlock(nn.Module):
288288

289289
def setup(self):
290290
logger.warning(
291-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
291+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
292292
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
293293
)
294294

@@ -381,7 +381,7 @@ class FlaxTransformer2DModel(nn.Module):
381381

382382
def setup(self):
383383
logger.warning(
384-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
384+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
385385
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
386386
)
387387

@@ -475,7 +475,7 @@ class FlaxFeedForward(nn.Module):
475475

476476
def setup(self):
477477
logger.warning(
478-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
478+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
479479
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
480480
)
481481

@@ -510,7 +510,7 @@ class FlaxGEGLU(nn.Module):
510510

511511
def setup(self):
512512
logger.warning(
513-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
513+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
514514
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
515515
)
516516

‎src/diffusers/models/controlnets/controlnet_flax.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ class FlaxControlNetConditioningEmbedding(nn.Module):
5454

5555
def setup(self) -> None:
5656
logger.warning(
57-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
57+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
5858
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
5959
)
6060

@@ -193,7 +193,7 @@ def init_weights(self, rng: jax.Array) -> FrozenDict:
193193

194194
def setup(self) -> None:
195195
logger.warning(
196-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
196+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
197197
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
198198
)
199199

‎src/diffusers/models/embeddings_flax.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ class FlaxTimestepEmbedding(nn.Module):
8282
"""
8383

8484
logger.warning(
85-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
85+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
8686
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
8787
)
8888

@@ -115,7 +115,7 @@ class FlaxTimesteps(nn.Module):
115115
freq_shift: float = 1
116116

117117
logger.warning(
118-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
118+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
119119
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
120120
)
121121

‎src/diffusers/models/modeling_flax_utils.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -291,7 +291,7 @@ def from_pretrained(
291291
```
292292
"""
293293
logger.warning(
294-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
294+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
295295
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
296296
)
297297
config = kwargs.pop("config", None)

‎src/diffusers/models/resnet_flax.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ class FlaxUpsample2D(nn.Module):
2727

2828
def setup(self):
2929
logger.warning(
30-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
30+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
3131
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
3232
)
3333

@@ -56,7 +56,7 @@ class FlaxDownsample2D(nn.Module):
5656

5757
def setup(self):
5858
logger.warning(
59-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
59+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
6060
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
6161
)
6262

@@ -84,7 +84,7 @@ class FlaxResnetBlock2D(nn.Module):
8484

8585
def setup(self):
8686
logger.warning(
87-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
87+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
8888
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
8989
)
9090

‎src/diffusers/models/unets/unet_2d_blocks_flax.py‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ class FlaxCrossAttnDownBlock2D(nn.Module):
6565

6666
def setup(self):
6767
logger.warning(
68-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
68+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
6969
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
7070
)
7171

@@ -145,7 +145,7 @@ class FlaxDownBlock2D(nn.Module):
145145

146146
def setup(self):
147147
logger.warning(
148-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
148+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
149149
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
150150
)
151151

@@ -223,7 +223,7 @@ class FlaxCrossAttnUpBlock2D(nn.Module):
223223

224224
def setup(self):
225225
logger.warning(
226-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
226+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
227227
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
228228
)
229229

@@ -308,7 +308,7 @@ class FlaxUpBlock2D(nn.Module):
308308

309309
def setup(self):
310310
logger.warning(
311-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
311+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
312312
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
313313
)
314314

@@ -381,7 +381,7 @@ class FlaxUNetMidBlock2DCrossAttn(nn.Module):
381381

382382
def setup(self):
383383
logger.warning(
384-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
384+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
385385
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
386386
)
387387

‎src/diffusers/models/unets/unet_2d_condition_flax.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -167,7 +167,7 @@ def init_weights(self, rng: jax.Array) -> FrozenDict:
167167

168168
def setup(self) -> None:
169169
logger.warning(
170-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
170+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
171171
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
172172
)
173173

‎src/diffusers/models/vae_flax.py‎

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ class FlaxUpsample2D(nn.Module):
7777

7878
def setup(self):
7979
logger.warning(
80-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
80+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
8181
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
8282
)
8383
self.conv = nn.Conv(
@@ -115,7 +115,7 @@ class FlaxDownsample2D(nn.Module):
115115

116116
def setup(self):
117117
logger.warning(
118-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
118+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
119119
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
120120
)
121121

@@ -162,7 +162,7 @@ class FlaxResnetBlock2D(nn.Module):
162162

163163
def setup(self):
164164
logger.warning(
165-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
165+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
166166
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
167167
)
168168

@@ -239,7 +239,7 @@ class FlaxAttentionBlock(nn.Module):
239239

240240
def setup(self):
241241
logger.warning(
242-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
242+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
243243
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
244244
)
245245

@@ -325,7 +325,7 @@ class FlaxDownEncoderBlock2D(nn.Module):
325325

326326
def setup(self):
327327
logger.warning(
328-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
328+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
329329
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
330330
)
331331

@@ -387,7 +387,7 @@ class FlaxUpDecoderBlock2D(nn.Module):
387387

388388
def setup(self):
389389
logger.warning(
390-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
390+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
391391
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
392392
)
393393

@@ -446,7 +446,7 @@ class FlaxUNetMidBlock2D(nn.Module):
446446

447447
def setup(self):
448448
logger.warning(
449-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
449+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
450450
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
451451
)
452452

@@ -542,7 +542,7 @@ class FlaxEncoder(nn.Module):
542542

543543
def setup(self):
544544
logger.warning(
545-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
545+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
546546
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
547547
)
548548

@@ -659,7 +659,7 @@ class FlaxDecoder(nn.Module):
659659

660660
def setup(self):
661661
logger.warning(
662-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
662+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
663663
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
664664
)
665665

@@ -836,7 +836,7 @@ class FlaxAutoencoderKL(nn.Module, FlaxModelMixin, ConfigMixin):
836836

837837
def setup(self):
838838
logger.warning(
839-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
839+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
840840
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
841841
)
842842

‎src/diffusers/pipelines/pipeline_flax_utils.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -313,7 +313,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
313313
```
314314
"""
315315
logger.warning(
316-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
316+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
317317
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
318318
)
319319

‎src/diffusers/schedulers/scheduling_utils_flax.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ def from_pretrained(
136136
137137
"""
138138
logger.warning(
139-
"Flax classes are deprecated and will be removed in Diffusers v1. We "
139+
"Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
140140
"recommend migrating to PyTorch classes or pinning your version of Diffusers."
141141
)
142142
config, kwargs = cls.load_config(

0 commit comments

Comments
 (0)