Update README.md

README.md

@@ -189,7 +189,7 @@ import torch
 pipeline = FluxPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16
 ).to("cuda")
-# Change this.
+# Change this accordingly.
 pipeline.load_lora_weights("optimus_16.safetensors")
 
 prompts = [
@@ -206,7 +206,7 @@ images = pipeline(
     generator=torch.manual_seed(0)
 ).images
 for i, image in enumerate(images):
-    image.save(f"{i}_{'upsampled' if upsample else 'non_upsampled'}.png")
+    image.save(f"{i}.png")
 ```
 
 </details>