Update README.md
#4
by
williamberman
- opened
README.md
CHANGED
@@ -9,7 +9,7 @@ Karlo is available in diffusers!
|
|
9 |
from diffusers import UnCLIPPipeline
|
10 |
import torch
|
11 |
|
12 |
-
pipe = UnCLIPPipeline.from_pretrained("
|
13 |
pipe = pipe.to('cuda')
|
14 |
|
15 |
prompt = "a high-resolution photograph of a big red frog on a green leaf."
|
|
|
9 |
from diffusers import UnCLIPPipeline
|
10 |
import torch
|
11 |
|
12 |
+
pipe = UnCLIPPipeline.from_pretrained("kakaobrain/karlo-v1-alpha", torch_dtype=torch.float16)
|
13 |
pipe = pipe.to('cuda')
|
14 |
|
15 |
prompt = "a high-resolution photograph of a big red frog on a green leaf."
|