Update README.md

README.md

@@ -21,6 +21,12 @@ widget: []
 
 FLUX.1 [dev] is a 12 billion parameter rectified flow transformer capable of generating images from text descriptions. 
 
+Install ```diffusers```
+
+```raw
+pip install -U diffusers
+```
+
 ## Model description
 
 These are LoRA adaption weights for the FLUX.1 [dev] model (```black-forest-labs/FLUX.1-dev```). This is a gated model, you must first get access to it before loading this LoRA adapter. 
@@ -90,12 +96,14 @@ repo_id = 'lamm-mit/leaf-FLUX.1-dev'
 pipeline = FluxPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev",
     torch_dtype=torch.bfloat16,
+    max_sequence_length=512,
 )
 
+#pipeline.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Comment out if you have enough GPU VRAM
+
 pipeline.load_lora_weights(repo_id, )
 pipeline=pipeline.to('cuda')
 ```
-
 Image generation:
 
 ```python
@@ -115,7 +123,9 @@ for _ in range(num_rows):
      
     all_images.extend(image)
 
-grid = image_grid(all_images, num_rows, num_samples,  ave_individual_files=True,  )
+grid = image_grid(all_images, num_rows, num_samples,
+                  save_individual_files=True,  )
 grid
 ```
 
+![Uploading image_grid_20240902_063338.png…]()