Update README.md
#1
by
zye2 - opened
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(
|