etri-vilab
/

koala-1b-llava-cap

StableDiffusionXLPipeline

Inference Endpoints

Model card Files Files and versions Community

ywlee88 commited on Jan 12

Commit

70ed488

•

1 Parent(s): 81127ce

update README

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -109,7 +109,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-llava-cap", torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 prompt = "A portrait painting of a Golden Retriever like Leonard da Vinci"