Update README.md

README.md

@@ -39,15 +39,14 @@ Example prompts:
 ### 2. Using Python Code
 
 ```python
-from diffusers import StableDiffusionPipeline
+from diffusers import StableDiffusion3Pipeline
 import torch
 
-pipeline = StableDiffusionPipeline.from_pretrained(
-    "CarlosRiverMe/SimpleTuner-stable-diffusion-3.5-large-ZukyTheDog",
-    torch_dtype=torch.float16
-)
-pipeline.to("cuda")
+pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large", torch_dtype=torch.bfloat16, variant = 'fp16', use_safetensors = True)
+
+pipe.to("cuda")
+pipe.load_lora_weights("CarlosRiverMe/SimpleTuner-stable-diffusion-3.5-large-ZukyTheDog", weight_name="pytorch_lora_weights.safetensors")
 
 prompt = "A photo of ZukyTheDog sitting in a garden"
-image = pipeline(prompt).images[0]
+image = pipe(prompt).images[0]
 image.save("generated_image.png")