Correct CogVideoX base reference
Browse files
README.md
CHANGED
|
@@ -134,7 +134,7 @@ The quantization pipeline is at [github.com/Asad-Ismail/ternary-models/tree/main
|
|
| 134 |
```python
|
| 135 |
# 1. Load pipeline
|
| 136 |
from diffusers import DiffusionPipeline
|
| 137 |
-
pipe = DiffusionPipeline.from_pretrained("
|
| 138 |
|
| 139 |
# 2. Extract DiT transformer
|
| 140 |
transformer = pipe.transformer
|
|
|
|
| 134 |
```python
|
| 135 |
# 1. Load pipeline
|
| 136 |
from diffusers import DiffusionPipeline
|
| 137 |
+
pipe = DiffusionPipeline.from_pretrained("zai-org/CogVideoX-2b", torch_dtype=torch.float16)
|
| 138 |
|
| 139 |
# 2. Extract DiT transformer
|
| 140 |
transformer = pipe.transformer
|