ywlee88 commited on
Commit
70ed488
1 Parent(s): 81127ce

update README

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -109,7 +109,7 @@ The inference code with denoising step 25
109
  import torch
110
  from diffusers import StableDiffusionXLPipeline
111
 
112
- pipe = StableDiffusionXLPipeline.from_pretrained("etri-vilab/koala-700m", torch_dtype=torch.float16)
113
  pipe = pipe.to("cuda")
114
 
115
  prompt = "A portrait painting of a Golden Retriever like Leonard da Vinci"
 
109
  import torch
110
  from diffusers import StableDiffusionXLPipeline
111
 
112
+ pipe = StableDiffusionXLPipeline.from_pretrained("etri-vilab/koala-1b-llava-cap", torch_dtype=torch.float16)
113
  pipe = pipe.to("cuda")
114
 
115
  prompt = "A portrait painting of a Golden Retriever like Leonard da Vinci"