animemory commited on
Commit
568835c
1 Parent(s): 309643a

readme update

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 AniMemoryPipeLine
122
  import torch
123
 
124
- pipe = AniMemoryPipeLine.from_pretrained("animEEEmpire/AniMemory-alpha", torch_dtype=torch.bfloat16)
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 = "一只凶恶的狼,猩红的眼神,在午夜咆哮,月光皎洁"