Commit
•
c8dc097
1
Parent(s):
5dba71c
Update README.md (#4)
Browse files- Update README.md (815e8ddfe5259c7ea7bbe2141f02d4ef2ce1feec)
Co-authored-by: Will Berman <[email protected]>
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."
|