Update examples/dreambooth/train_dreambooth_lora_flux_kontext.py

Co-authored-by: Aryan <aryan@huggingface.co>

Author: sayakpaul

examples/dreambooth/train_dreambooth_lora_flux_kontext.py

@@ -127,7 +127,7 @@ def save_model_card(
 ```py
 from diffusers import FluxKontextPipeline
 import torch
-pipeline = FluxKontextPipeline.from_pretrained("black-forest-labs/KONTEXT.1-dev", torch_dtype=torch.bfloat16).to('cuda')
+pipeline = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to('cuda')
 pipeline.load_lora_weights('{repo_id}', weight_name='pytorch_lora_weights.safetensors')
 image = pipeline('{validation_prompt if validation_prompt else instance_prompt}').images[0]
 ```