Update README.md
Browse files
README.md
CHANGED
@@ -36,12 +36,15 @@ weight_type = torch.float32 # or float16
|
|
36 |
|
37 |
# Load model.
|
38 |
pipe = StableDiffusionPipeline.from_pretrained(repo, torch_dtype=weight_type)
|
39 |
-
|
|
|
|
|
|
|
40 |
pipe.to("cuda")
|
41 |
|
42 |
prompt = "portrait photo of a girl, photograph, highly detailed face, depth of field, moody light, golden hour"
|
43 |
|
44 |
-
# Ensure using
|
45 |
image = pipe(
|
46 |
prompt,
|
47 |
num_inference_steps=1,
|
|
|
36 |
|
37 |
# Load model.
|
38 |
pipe = StableDiffusionPipeline.from_pretrained(repo, torch_dtype=weight_type)
|
39 |
+
|
40 |
+
# use original VAE
|
41 |
+
# pipe.vae = AutoencoderKL.from_pretrained("IDKiro/sdxs-512-0.9/vae_large")
|
42 |
+
|
43 |
pipe.to("cuda")
|
44 |
|
45 |
prompt = "portrait photo of a girl, photograph, highly detailed face, depth of field, moody light, golden hour"
|
46 |
|
47 |
+
# Ensure using 1 inference step and CFG set to 0.
|
48 |
image = pipe(
|
49 |
prompt,
|
50 |
num_inference_steps=1,
|