etri-vilab
/

koala-1b

StableDiffusionXLPipeline

Inference Endpoints

Model card Files Files and versions Community

ywlee88 commited on Jan 12

Commit

23cde0b

•

1 Parent(s): 7de8133

update README

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -105,7 +105,7 @@ The inference code with denoising step 25
 import torch
 from diffusers import StableDiffusionXLPipeline
-pipe = StableDiffusionXLPipeline.from_pretrained("etri-vilab/koala-700m", torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 prompt = "A portrait painting of a Golden Retriever like Leonard da Vinci"

 import torch
 from diffusers import StableDiffusionXLPipeline
+pipe = StableDiffusionXLPipeline.from_pretrained("etri-vilab/koala-1b", torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 prompt = "A portrait painting of a Golden Retriever like Leonard da Vinci"