Files changed (1) hide show
  1. README.md +6 -0
README.md CHANGED
@@ -22,7 +22,13 @@ Feel free to request for other models for compression as well (for either the `d
22
  ```python
23
  import torch
24
  from diffusers import LongCatImageEditPipeline, LongCatImageTransformer2DModel
 
 
 
 
 
25
  from transformers.modeling_utils import no_init_weights
 
26
  with no_init_weights():
27
  transformer = LongCatImageTransformer2DModel.from_config(
28
  LongCatImageTransformer2DModel.load_config(
 
22
  ```python
23
  import torch
24
  from diffusers import LongCatImageEditPipeline, LongCatImageTransformer2DModel
25
+
26
+ # for transformers version >=5.0.0
27
+ # from transformers.initialization import no_init_weights
28
+
29
+ # else
30
  from transformers.modeling_utils import no_init_weights
31
+
32
  with no_init_weights():
33
  transformer = LongCatImageTransformer2DModel.from_config(
34
  LongCatImageTransformer2DModel.load_config(