etri-vilab
/

koala-lightning-1b

StableDiffusionXLPipeline

Inference Endpoints

Model card Files Files and versions Community

ywlee88 commited on May 29, 2024

Commit

65b52b4

·

1 Parent(s): 47232f6

update README

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -112,7 +112,7 @@ The inference code with denoising step 25
 import torch
 from diffusers import StableDiffusionXLPipeline, EulerDiscreteScheduler
-pipe = StableDiffusionXLPipeline.from_pretrained("etri-vilab/koala-lightning-700m", torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 # Ensure sampler uses "trailing" timesteps and "sample" prediction type.

 import torch
 from diffusers import StableDiffusionXLPipeline, EulerDiscreteScheduler
+pipe = StableDiffusionXLPipeline.from_pretrained("etri-vilab/koala-lightning-1b", torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 # Ensure sampler uses "trailing" timesteps and "sample" prediction type.