Update README.md

README.md

@@ -11,7 +11,7 @@ import torch, torchvision
 from diffusers import StableDiffusionPipeline, UNet2DConditionModel
 from src.utils_perflow import merge_delta_weights_into_unet
 from src.scheduler_perflow import PeRFlowScheduler
-delta_weights = UNet2DConditionModel.from_pretrained("hansyan/piecewise-rectified-flow-delta-weights", torch_dtype=torch.float16, variant="v0-1",).state_dict()
+delta_weights = UNet2DConditionModel.from_pretrained("hansyan/perflow-sd15-delta-weights", torch_dtype=torch.float16, variant="v0-1",).state_dict()
 pipe = StableDiffusionPipeline.from_pretrained("Lykon/dreamshaper-8", torch_dtype=torch.float16,)
 pipe = merge_delta_weights_into_unet(pipe, delta_weights)
 pipe.scheduler = PeRFlowScheduler.from_config(pipe.scheduler.config, prediction_type="epsilon", num_time_windows=4)