| import torch | |
| from diffusers import StableDiffusion3Pipeline | |
| from diffusers import AutoPipelineForText2Image | |
| from safetensors.torch import load_file | |
| torch.backends.cuda.enable_mem_efficient_sdp(False) | |
| torch.backends.cuda.enable_flash_sdp(False) | |
| pipeline = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16).to('cuda') | |
| pipeline.load_lora_weights("rdeinla/test-can-1-2", weight_name='pytorch_lora_weights.safetensors') | |
| image = pipeline("A photo of a young canola plant with medium-sized leaves. The leaves are ruffled around the edges and have prominent ridges. It is about 19 days old. It is growing in dark, nutrient-rich soil. It is contained within a smooth, bright blue cylindrical cup on a bright blue background", | |
| guidance_scale = 5.0, negative_prompt = "blurry").images[0] | |
| image.save("canola_12_5_1.png") |