readme update
Browse files
README.md
CHANGED
|
@@ -118,10 +118,10 @@ pip install .
|
|
| 118 |
- And then, you can use the following code to generate images.
|
| 119 |
|
| 120 |
```python
|
| 121 |
-
from diffusers import
|
| 122 |
import torch
|
| 123 |
|
| 124 |
-
pipe =
|
| 125 |
pipe.to("cuda")
|
| 126 |
|
| 127 |
prompt = "一只凶恶的狼,猩红的眼神,在午夜咆哮,月光皎洁"
|
|
|
|
| 118 |
- And then, you can use the following code to generate images.
|
| 119 |
|
| 120 |
```python
|
| 121 |
+
from diffusers import AniMemoryPipeline
|
| 122 |
import torch
|
| 123 |
|
| 124 |
+
pipe = AniMemoryPipeline.from_pretrained("animEEEmpire/AniMemory-alpha", torch_dtype=torch.bfloat16)
|
| 125 |
pipe.to("cuda")
|
| 126 |
|
| 127 |
prompt = "一只凶恶的狼,猩红的眼神,在午夜咆哮,月光皎洁"
|