Update README.md (#1)
Browse files- Update README.md (074c05cde42a9415a11827ddb87862f6c467bb40)
- Update README.md (f92ad15a5282432b8ae794c0ccc3c144231af28d)
Co-authored-by: zy <zye2@users.noreply.huggingface.co>
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(
|