hansyan
/

perflow-sd15-delta-weights

Model card Files Files and versions Community

hansyan commited on Mar 7

Commit

5c0c439

•

1 Parent(s): 70fdf0d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -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()
 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)

 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/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)