Commit
•
c8dc097
1
Parent(s):
5dba71c
Update README.md (#4)
Browse files- Update README.md (815e8ddfe5259c7ea7bbe2141f02d4ef2ce1feec)
Co-authored-by: Will Berman <williamberman@users.noreply.huggingface.co>
README.md
CHANGED
|
@@ -22,7 +22,7 @@ pip install diffusers transformers accelerate safetensors
|
|
| 22 |
from diffusers import UnCLIPPipeline
|
| 23 |
import torch
|
| 24 |
|
| 25 |
-
pipe = UnCLIPPipeline.from_pretrained("
|
| 26 |
pipe = pipe.to('cuda')
|
| 27 |
|
| 28 |
prompt = "a high-resolution photograph of a big red frog on a green leaf."
|
|
|
|
| 22 |
from diffusers import UnCLIPPipeline
|
| 23 |
import torch
|
| 24 |
|
| 25 |
+
pipe = UnCLIPPipeline.from_pretrained("kakaobrain/karlo-v1-alpha", torch_dtype=torch.float16)
|
| 26 |
pipe = pipe.to('cuda')
|
| 27 |
|
| 28 |
prompt = "a high-resolution photograph of a big red frog on a green leaf."
|