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 = "一只凶恶的狼,猩红的眼神,在午夜咆哮,月光皎洁"
|