Update README.md

README.md

@@ -22,39 +22,6 @@ These are the control weights trained on [black-forest-labs/FLUX.1-dev](htpss://
 and [TIGER-Lab/OmniEdit-Filtered-1.2M](https://huggingface.co/datasets/TIGER-Lab/OmniEdit-Filtered-1.2M) for image editing. We use the
 [Flux Control framework](https://blackforestlabs.ai/flux-1-tools/) for fine-tuning.
 
-prompt: Give this the look of a traditional Japanese woodblock print.
-![images_0)](./images_0.png)
-prompt: Give this the look of a traditional Japanese woodblock print.
-![images_1)](./images_1.png)
-prompt: Give this the look of a traditional Japanese woodblock print.
-![images_2)](./images_2.png)
-prompt: Give this the look of a traditional Japanese woodblock print.
-![images_3)](./images_3.png)
-prompt: transform the setting to a winter scene
-![images_4)](./images_4.png)
-prompt: transform the setting to a winter scene
-![images_5)](./images_5.png)
-prompt: transform the setting to a winter scene
-![images_6)](./images_6.png)
-prompt: transform the setting to a winter scene
-![images_7)](./images_7.png)
-prompt: Change it to look like it's in the style of an impasto painting.
-![images_8)](./images_8.png)
-prompt: Change it to look like it's in the style of an impasto painting.
-![images_9)](./images_9.png)
-prompt: Change it to look like it's in the style of an impasto painting.
-![images_10)](./images_10.png)
-prompt: Change it to look like it's in the style of an impasto painting.
-![images_11)](./images_11.png)
-prompt: turn the color of mushroom to gray
-![images_12)](./images_12.png)
-prompt: turn the color of mushroom to gray
-![images_13)](./images_13.png)
-prompt: turn the color of mushroom to gray
-![images_14)](./images_14.png)
-prompt: turn the color of mushroom to gray
-![images_15)](./images_15.png)
-
 
 ## License
 
@@ -110,15 +77,15 @@ import torch
 
 path = "sayakpaul/FLUX.1-dev-edit-v0" # to change
 edit_transformer = FluxTransformer2DModel.from_pretrained(path, torch_dtype=torch.bfloat16)
-control_pipe = FluxControlPipeline.from_pretrained(
+pipeline = FluxControlPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev", transformer=edit_transformer, torch_dtype=torch.bfloat16
 ).to("cuda")
 
 # load the turbo LoRA
-control_pipe.load_lora_weights(
+pipeline.load_lora_weights(
     hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), adapter_name="hyper-sd"
 )
-control_pipe.set_adapters(["hyper-sd"], adapter_weights=[0.125])
+pipeline.set_adapters(["hyper-sd"], adapter_weights=[0.125])
 
 image = load_image("./assets/mushroom.jpg") # resize as needed.
 print(image.size)
@@ -185,30 +152,30 @@ You can also choose to perform quantization if the memory requirements cannot be
       <img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/assets/car.jpg" alt="Source Image 1" width="150"><br>
       <em>Give this the look of a traditional Japanese woodblock print.</em>
     </td>
-    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-10_car.jpg" alt="Edited Image gs 10" width="150"></td>
-    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-20_car.jpg" alt="Edited Image gs 20" width="150"></td>
-    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-30_car.jpg" alt="Edited Image gs 30" width="150"></td>
-    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-40_car.jpg" alt="Edited Image gs 40" width="150"></td>
+    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-10_car.jpg" alt="Edited Image gs 10"></td>
+    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-20_car.jpg" alt="Edited Image gs 20"></td>
+    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-30_car.jpg" alt="Edited Image gs 30"></td>
+    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-40_car.jpg" alt="Edited Image gs 40"></td>
   </tr>
   <tr>
     <td align="center">
       <img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/assets/green_creature" alt="Source Image 2" width="150"><br>
       <em>transform the setting to a winter scene</em>
     </td>
-    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-10_green_creature.jpg" alt="Edited Image gs 10" width="150"></td>
-    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-20_green_creature.jpg" alt="Edited Image gs 20" width="150"></td>
-    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-30_green_creature.jpg" alt="Edited Image gs 30" width="150"></td>
-    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-40_green_creature.jpg" alt="Edited Image gs 40" width="150"></td>
+    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-10_green_creature.jpg" alt="Edited Image gs 10"></td>
+    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-20_green_creature.jpg" alt="Edited Image gs 20"></td>
+    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-30_green_creature.jpg" alt="Edited Image gs 30"></td>
+    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-40_green_creature.jpg" alt="Edited Image gs 40"></td>
   </tr>
   <tr>
     <td align="center">
       <img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/assets/mushroom.jpg" alt="Source Image 3" width="150"><br>
       <em>turn the color of mushroom to gray</em>
     </td>
-    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-10_mushroom.jpg" alt="Edited Image gs 10" width="150"></td>
-    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-20_mushroom.jpg" alt="Edited Image gs 20" width="150"></td>
-    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-30_mushroom.jpg" alt="Edited Image gs 30" width="150"></td>
-    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-40_mushroom.jpg" alt="Edited Image gs 40" width="150"></td>
+    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-10_mushroom.jpg" alt="Edited Image gs 10"></td>
+    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-20_mushroom.jpg" alt="Edited Image gs 20"></td>
+    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-30_mushroom.jpg" alt="Edited Image gs 30"></td>
+    <td align="center"><img src="https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_gs-40_mushroom.jpg" alt="Edited Image gs 40"></td>
   </tr>
 </table>