Update README.md
Browse files
    	
        README.md
    CHANGED
    
    | @@ -8,7 +8,7 @@ language: | |
| 8 | 
             
            pipeline_tag: text-to-image
         | 
| 9 | 
             
            library_name: diffusers
         | 
| 10 | 
             
            ---
         | 
| 11 | 
            -
            For more information (including how to compress models yourself), check out https://huggingface.co/DFloat11
         | 
| 12 |  | 
| 13 | 
             
            This is my first time using DF11 to compress a model outside the Flux architecture. The process for compressing Flux-based models is much more straightforward as compared to other architectures because the compression code requires a `pattern_dict` as input, but the original [example code](https://github.com/LeanModels/DFloat11/tree/master/examples/compress_flux1) only provides it for Flux, which meant I had to learn the notation myself and modify it to fit other models. At least Chroma is just a pruned version of Flux, so it was relatively simple to derive the correct `pattern_dict` this time. Do let me know if you run into any problems.
         | 
| 14 |  | 
| @@ -65,7 +65,7 @@ pattern_dict = { | |
| 65 | 
             
                        torch_dtype=torch.bfloat16
         | 
| 66 | 
             
                    ).to(torch.bfloat16)
         | 
| 67 |  | 
| 68 | 
            -
                pipe =  | 
| 69 | 
             
                    "lodestones/Chroma1-HD",
         | 
| 70 | 
             
                    transformer=transformer,
         | 
| 71 | 
             
                    torch_dtype=torch.bfloat16
         | 
|  | |
| 8 | 
             
            pipeline_tag: text-to-image
         | 
| 9 | 
             
            library_name: diffusers
         | 
| 10 | 
             
            ---
         | 
| 11 | 
            +
            For more information (including how to compress models yourself), check out https://huggingface.co/DFloat11 and https://github.com/LeanModels/DFloat11
         | 
| 12 |  | 
| 13 | 
             
            This is my first time using DF11 to compress a model outside the Flux architecture. The process for compressing Flux-based models is much more straightforward as compared to other architectures because the compression code requires a `pattern_dict` as input, but the original [example code](https://github.com/LeanModels/DFloat11/tree/master/examples/compress_flux1) only provides it for Flux, which meant I had to learn the notation myself and modify it to fit other models. At least Chroma is just a pruned version of Flux, so it was relatively simple to derive the correct `pattern_dict` this time. Do let me know if you run into any problems.
         | 
| 14 |  | 
|  | |
| 65 | 
             
                        torch_dtype=torch.bfloat16
         | 
| 66 | 
             
                    ).to(torch.bfloat16)
         | 
| 67 |  | 
| 68 | 
            +
                pipe = ChromaPipeline.from_pretrained(
         | 
| 69 | 
             
                    "lodestones/Chroma1-HD",
         | 
| 70 | 
             
                    transformer=transformer,
         | 
| 71 | 
             
                    torch_dtype=torch.bfloat16
         |