| from diffsynth import ModelManager, SDImagePipeline, download_models |
| import torch |
|
|
|
|
| |
| |
| |
| |
| |
| download_models(["AingDiffusion_v12", "IP-Adapter-SD", "TextualInversion_VeryBadImageNegative_v1.3"]) |
|
|
| |
| model_manager = ModelManager(torch_dtype=torch.float16, device="cuda") |
| model_manager.load_models([ |
| "models/stable_diffusion/aingdiffusion_v12.safetensors", |
| "models/IpAdapter/stable_diffusion/image_encoder/model.safetensors", |
| "models/IpAdapter/stable_diffusion/ip-adapter_sd15.bin" |
| ]) |
| pipe = SDImagePipeline.from_model_manager(model_manager) |
| pipe.prompter.load_textual_inversions(["models/textual_inversion/verybadimagenegative_v1.3.pt"]) |
|
|
| torch.manual_seed(1) |
| style_image = pipe( |
| prompt="masterpiece, best quality, a car", |
| negative_prompt="verybadimagenegative_v1.3", |
| cfg_scale=7, clip_skip=2, |
| height=512, width=512, num_inference_steps=50, |
| ) |
| style_image.save("car.jpg") |
|
|
| image = pipe( |
| prompt="masterpiece, best quality, a car running on the road", |
| negative_prompt="verybadimagenegative_v1.3", |
| cfg_scale=7, clip_skip=2, |
| height=512, width=512, num_inference_steps=50, |
| ipadapter_images=[style_image], ipadapter_scale=1.0 |
| ) |
| image.save("car_on_the_road.jpg") |
|
|