Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	Upload 2 files
Browse files- app.py +4 -2
- loras.json +19 -7
    	
        app.py
    CHANGED
    
    | @@ -43,6 +43,7 @@ last_model = models[0] | |
| 43 | 
             
            last_cn_on = False
         | 
| 44 | 
             
            #controlnet_union = FluxControlNetModel.from_pretrained(controlnet_model_union_repo, torch_dtype=dtype)
         | 
| 45 | 
             
            #controlnet = FluxMultiControlNetModel([controlnet_union])
         | 
|  | |
| 46 |  | 
| 47 | 
             
            MAX_SEED = 2**32-1
         | 
| 48 |  | 
| @@ -66,6 +67,7 @@ def change_base_model(repo_id: str, cn_on: bool, disable_model_cache: bool, prog | |
| 66 | 
             
                        print(f"Loading model: {repo_id} / Loading ControlNet: {controlnet_model_union_repo}")
         | 
| 67 | 
             
                        controlnet_union = FluxControlNetModel.from_pretrained(controlnet_model_union_repo, torch_dtype=dtype)
         | 
| 68 | 
             
                        controlnet = FluxMultiControlNetModel([controlnet_union])
         | 
|  | |
| 69 | 
             
                        pipe = FluxControlNetPipeline.from_pretrained(repo_id, controlnet=controlnet, torch_dtype=dtype)
         | 
| 70 | 
             
                        pipe_i2i = FluxControlNetImg2ImgPipeline.from_pretrained(repo_id, controlnet=controlnet, vae=None, transformer=pipe.transformer, text_encoder=pipe.text_encoder,
         | 
| 71 | 
             
                         tokenizer=pipe.tokenizer, text_encoder_2=pipe.text_encoder_2, tokenizer_2=pipe.tokenizer_2, torch_dtype=dtype)
         | 
| @@ -194,7 +196,7 @@ def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps | |
| 194 | 
             
                    with calculateDuration("Generating image"):
         | 
| 195 | 
             
                        # Generate image
         | 
| 196 | 
             
                        modes, images, scales = get_control_params()
         | 
| 197 | 
            -
                        if  | 
| 198 | 
             
                            pipe_i2i.to("cuda")
         | 
| 199 | 
             
                            pipe_i2i.vae = good_vae
         | 
| 200 | 
             
                            image_input = load_image(image_input_path)
         | 
| @@ -212,7 +214,7 @@ def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps | |
| 212 | 
             
                                output_type="pil",
         | 
| 213 | 
             
                            ).images[0]
         | 
| 214 | 
             
                            return final_image 
         | 
| 215 | 
            -
                        else: | 
| 216 | 
             
                            pipe_i2i.to("cuda")
         | 
| 217 | 
             
                            pipe_i2i.vae = good_vae
         | 
| 218 | 
             
                            image_input = load_image(image_input_path)
         | 
|  | |
| 43 | 
             
            last_cn_on = False
         | 
| 44 | 
             
            #controlnet_union = FluxControlNetModel.from_pretrained(controlnet_model_union_repo, torch_dtype=dtype)
         | 
| 45 | 
             
            #controlnet = FluxMultiControlNetModel([controlnet_union])
         | 
| 46 | 
            +
            #controlnet.config = controlnet_union.config
         | 
| 47 |  | 
| 48 | 
             
            MAX_SEED = 2**32-1
         | 
| 49 |  | 
|  | |
| 67 | 
             
                        print(f"Loading model: {repo_id} / Loading ControlNet: {controlnet_model_union_repo}")
         | 
| 68 | 
             
                        controlnet_union = FluxControlNetModel.from_pretrained(controlnet_model_union_repo, torch_dtype=dtype)
         | 
| 69 | 
             
                        controlnet = FluxMultiControlNetModel([controlnet_union])
         | 
| 70 | 
            +
                        controlnet.config = controlnet_union.config
         | 
| 71 | 
             
                        pipe = FluxControlNetPipeline.from_pretrained(repo_id, controlnet=controlnet, torch_dtype=dtype)
         | 
| 72 | 
             
                        pipe_i2i = FluxControlNetImg2ImgPipeline.from_pretrained(repo_id, controlnet=controlnet, vae=None, transformer=pipe.transformer, text_encoder=pipe.text_encoder,
         | 
| 73 | 
             
                         tokenizer=pipe.tokenizer, text_encoder_2=pipe.text_encoder_2, tokenizer_2=pipe.tokenizer_2, torch_dtype=dtype)
         | 
|  | |
| 196 | 
             
                    with calculateDuration("Generating image"):
         | 
| 197 | 
             
                        # Generate image
         | 
| 198 | 
             
                        modes, images, scales = get_control_params()
         | 
| 199 | 
            +
                        if not cn_on or len(modes) == 0:
         | 
| 200 | 
             
                            pipe_i2i.to("cuda")
         | 
| 201 | 
             
                            pipe_i2i.vae = good_vae
         | 
| 202 | 
             
                            image_input = load_image(image_input_path)
         | 
|  | |
| 214 | 
             
                                output_type="pil",
         | 
| 215 | 
             
                            ).images[0]
         | 
| 216 | 
             
                            return final_image 
         | 
| 217 | 
            +
                        else:
         | 
| 218 | 
             
                            pipe_i2i.to("cuda")
         | 
| 219 | 
             
                            pipe_i2i.vae = good_vae
         | 
| 220 | 
             
                            image_input = load_image(image_input_path)
         | 
    	
        loras.json
    CHANGED
    
    | @@ -20,13 +20,6 @@ | |
| 20 | 
             
                "weights": "lora_pastel_anime_flux.safetensors",
         | 
| 21 | 
             
                "trigger_word": ""
         | 
| 22 | 
             
              },
         | 
| 23 | 
            -
              {
         | 
| 24 | 
            -
                "image": "https://huggingface.co/Norod78/JojosoStyle-flux-lora/resolve/main/samples/1725244218477__000004255_1.jpg",
         | 
| 25 | 
            -
                "title": "JoJo's Bizarre Adventure: Stone Ocean ",
         | 
| 26 | 
            -
                "repo": "Norod78/JojosoStyle-flux-lora",
         | 
| 27 | 
            -
                "weights": "JojosoStyle_flux_lora.safetensors",
         | 
| 28 | 
            -
                "trigger_word": ", JojosoStyle"
         | 
| 29 | 
            -
              },
         | 
| 30 | 
             
              {
         | 
| 31 | 
             
                "image": "https://huggingface.co/prithivMLmods/Canopus-Anime-Art-Flux-LoRA/resolve/main/images/BBBB.png",
         | 
| 32 | 
             
                "title": "Canopus-Anime-Art-FluxDev-LoRA ",
         | 
| @@ -41,6 +34,25 @@ | |
| 41 | 
             
                "weights": "ovld_style_overlord_wavymulder.safetensors",
         | 
| 42 | 
             
                "trigger_word": ", ovld style anime"
         | 
| 43 | 
             
              },
         | 
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
| 44 | 
             
              {
         | 
| 45 | 
             
                "image": "https://huggingface.co/multimodalart/flux-tarot-v1/resolve/main/images/e5f2761e5d474e6ba492d20dca0fa26f_e78f1524074b42b6ac49643ffad50ac6.png",
         | 
| 46 | 
             
                "title": "Tarot v1",
         | 
|  | |
| 20 | 
             
                "weights": "lora_pastel_anime_flux.safetensors",
         | 
| 21 | 
             
                "trigger_word": ""
         | 
| 22 | 
             
              },
         | 
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
| 23 | 
             
              {
         | 
| 24 | 
             
                "image": "https://huggingface.co/prithivMLmods/Canopus-Anime-Art-Flux-LoRA/resolve/main/images/BBBB.png",
         | 
| 25 | 
             
                "title": "Canopus-Anime-Art-FluxDev-LoRA ",
         | 
|  | |
| 34 | 
             
                "weights": "ovld_style_overlord_wavymulder.safetensors",
         | 
| 35 | 
             
                "trigger_word": ", ovld style anime"
         | 
| 36 | 
             
              },
         | 
| 37 | 
            +
              {
         | 
| 38 | 
            +
                "image": "https://huggingface.co/renderartist/simplevectorflux/resolve/main/images/ComfyUI_09477_.jpeg",
         | 
| 39 | 
            +
                "title": "Simple Vector",
         | 
| 40 | 
            +
                "repo": "renderartist/simplevectorflux",
         | 
| 41 | 
            +
                "trigger_word": "v3ct0r style, simple flat vector art, isolated on white bg,",
         | 
| 42 | 
            +
                "trigger_position": "prepend"
         | 
| 43 | 
            +
              },
         | 
| 44 | 
            +
              {
         | 
| 45 | 
            +
                "image": "https://huggingface.co/glif/how2draw/resolve/main/images/glif-how2draw-araminta-k-vbnvy94npt8m338r2vm02m50.jpg",
         | 
| 46 | 
            +
                "repo": "glif/how2draw",
         | 
| 47 | 
            +
                "trigger_word": ", How2Draw",
         | 
| 48 | 
            +
                "title": "How2Draw"
         | 
| 49 | 
            +
              },
         | 
| 50 | 
            +
              {
         | 
| 51 | 
            +
                "image": "https://huggingface.co/glif/anime-blockprint-style/resolve/main/images/glif-block-print-anime-flux-dev-araminta-k-lora-araminta-k-kbde06qyovrmvsv65ubfyhn1.jpg",
         | 
| 52 | 
            +
                "repo": "glif/anime-blockprint-style",
         | 
| 53 | 
            +
                "trigger_word": ", blockprint style",
         | 
| 54 | 
            +
                "title": "Blockprint Style"
         | 
| 55 | 
            +
              },
         | 
| 56 | 
             
              {
         | 
| 57 | 
             
                "image": "https://huggingface.co/multimodalart/flux-tarot-v1/resolve/main/images/e5f2761e5d474e6ba492d20dca0fa26f_e78f1524074b42b6ac49643ffad50ac6.png",
         | 
| 58 | 
             
                "title": "Tarot v1",
         | 
 
			
