IDKiro commited on
Commit
36a3735
1 Parent(s): b980f65

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -2
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
- # pipe.vae = AutoencoderKL.from_pretrained("IDKiro/sdxs-512-0.9/vae_large") # use original VAE
 
 
 
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 the same inference steps as the loaded model and CFG set to 0.
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,