Update README.md

README.md

@@ -8,7 +8,7 @@ tags:
 - safetensors
 ---
 
-# Model Card for sdiazlor/flux2-klein-4b-optimized-smashed
+# Model Card for PrunaAI/flux2-klein-4b-optimized-smashed
 
 This model was created using the [pruna](https://github.com/PrunaAI/pruna) library. Pruna is a model optimization framework built for developers, enabling you to deliver more efficient models with minimal implementation overhead.
 
@@ -20,7 +20,7 @@ First things first, you need to install the pruna library:
 pip install pruna
 ```
 
-You can [use the library_name library to load the model](https://huggingface.co/sdiazlor/flux2-klein-4b-optimized-smashed?library=library_name) but this might not include all optimizations by default.
+You can [use the library_name library to load the model](https://huggingface.co/PrunaAI/flux2-klein-4b-optimized-smashed?library=library_name) but this might not include all optimizations by default.
 
 To ensure that all optimizations are applied, use the pruna library to load the model using the following code:
 
@@ -28,7 +28,7 @@ To ensure that all optimizations are applied, use the pruna library to load the
 from pruna import PrunaModel
 
 loaded_model = PrunaModel.from_pretrained(
-    "sdiazlor/flux2-klein-4b-optimized-smashed"
+    "PrunaAI/flux2-klein-4b-optimized-smashed"
 )
 # we can then run inference using the methods supported by the base model
 ```