Update README.md
Browse files
README.md
CHANGED
@@ -43,7 +43,7 @@ unet = UNet2DConditionModel.from_pretrained(unet_path, torch_dtype=torch.float16
|
|
43 |
pipe = StableDiffusionXLPipeline.from_pretrained(base_path, unet=unet, torch_dtype=torch.float16, variant="fp16").to("cuda")
|
44 |
pipe.scheduler = EulerDiscreteScheduler.from_pretrained(base_path, subfolder="scheduler", torch_dtype=torch.float16)
|
45 |
|
46 |
-
refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(
|
47 |
|
48 |
# Generate image.
|
49 |
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"
|
|
|
43 |
pipe = StableDiffusionXLPipeline.from_pretrained(base_path, unet=unet, torch_dtype=torch.float16, variant="fp16").to("cuda")
|
44 |
pipe.scheduler = EulerDiscreteScheduler.from_pretrained(base_path, subfolder="scheduler", torch_dtype=torch.float16)
|
45 |
|
46 |
+
refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(refiner_path,torch_dtype=torch.float16, use_safetensors=True).to("cuda") # we found use base_path instead of refiner_path may get a better performance
|
47 |
|
48 |
# Generate image.
|
49 |
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"
|