cosmicman commited on
Commit
e88adcc
1 Parent(s): 1c33a2c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -32,8 +32,7 @@ unet = UNet2DConditionModel.from_pretrained(unet_path, torch_dtype=torch.float16
32
  pipe = StableDiffusionXLPipeline.from_pretrained(base_path, unet=unet, torch_dtype=torch.float16, variant="fp16").to("cuda")
33
  pipe.scheduler = EulerDiscreteScheduler.from_pretrained(base_path, subfolder="scheduler", torch_dtype=torch.float16)
34
 
35
- refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(refiner_path,torch_dtype=torch.float16, use_safetensors=True).to("cuda")
36
- refiner.scheduler = EulerDiscreteScheduler.from_pretrained(base_path, subfolder="scheduler", torch_dtype=torch.float16)
37
 
38
  # Generate image.
39
  positive_prompt = "A fit Caucasian elderly woman, her wavy white hair above shoulders, wears a pink floral cotton long-sleeve shirt and a cotton hat against a natural landscape in an upper body shot"
 
32
  pipe = StableDiffusionXLPipeline.from_pretrained(base_path, unet=unet, torch_dtype=torch.float16, variant="fp16").to("cuda")
33
  pipe.scheduler = EulerDiscreteScheduler.from_pretrained(base_path, subfolder="scheduler", torch_dtype=torch.float16)
34
 
35
+ refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(base_path,torch_dtype=torch.float16, use_safetensors=True).to("cuda") # using "base_path" instead of "refiner_path" may have a better result
 
36
 
37
  # Generate image.
38
  positive_prompt = "A fit Caucasian elderly woman, her wavy white hair above shoulders, wears a pink floral cotton long-sleeve shirt and a cotton hat against a natural landscape in an upper body shot"