Spaces:
Runtime error
Runtime error
| import os | |
| import json | |
| import copy | |
| import time | |
| import random | |
| import logging | |
| import numpy as np | |
| from typing import Any, Dict, List, Optional, Union | |
| import torch | |
| from PIL import Image | |
| import gradio as gr | |
| from diffusers import ( | |
| DiffusionPipeline, | |
| AutoencoderTiny, | |
| AutoencoderKL, | |
| AutoPipelineForImage2Image, | |
| FluxPipeline, | |
| FlowMatchEulerDiscreteScheduler) | |
| from huggingface_hub import ( | |
| hf_hub_download, | |
| HfFileSystem, | |
| ModelCard, | |
| snapshot_download) | |
| import spaces | |
| #---if env = local or colab--- | |
| # Authenticate with Hugging Face | |
| # from huggingface_hub import login | |
| # Log in to Hugging Face using the provided token | |
| # hf_token = 'hf-token-authentication' | |
| # login(hf_token) | |
| def calculate_shift( | |
| image_seq_len, | |
| base_seq_len: int = 256, | |
| max_seq_len: int = 4096, | |
| base_shift: float = 0.5, | |
| max_shift: float = 1.16, | |
| ): | |
| m = (max_shift - base_shift) / (max_seq_len - base_seq_len) | |
| b = base_shift - m * base_seq_len | |
| mu = image_seq_len * m + b | |
| return mu | |
| def retrieve_timesteps( | |
| scheduler, | |
| num_inference_steps: Optional[int] = None, | |
| device: Optional[Union[str, torch.device]] = None, | |
| timesteps: Optional[List[int]] = None, | |
| sigmas: Optional[List[float]] = None, | |
| **kwargs, | |
| ): | |
| if timesteps is not None and sigmas is not None: | |
| raise ValueError("Only one of `timesteps` or `sigmas` can be passed. Please choose one to set custom values") | |
| if timesteps is not None: | |
| scheduler.set_timesteps(timesteps=timesteps, device=device, **kwargs) | |
| timesteps = scheduler.timesteps | |
| num_inference_steps = len(timesteps) | |
| elif sigmas is not None: | |
| scheduler.set_timesteps(sigmas=sigmas, device=device, **kwargs) | |
| timesteps = scheduler.timesteps | |
| num_inference_steps = len(timesteps) | |
| else: | |
| scheduler.set_timesteps(num_inference_steps, device=device, **kwargs) | |
| timesteps = scheduler.timesteps | |
| return timesteps, num_inference_steps | |
| # FLUX pipeline | |
| def flux_pipe_call_that_returns_an_iterable_of_images( | |
| self, | |
| prompt: Union[str, List[str]] = None, | |
| prompt_2: Optional[Union[str, List[str]]] = None, | |
| height: Optional[int] = None, | |
| width: Optional[int] = None, | |
| num_inference_steps: int = 28, | |
| timesteps: List[int] = None, | |
| guidance_scale: float = 3.5, | |
| num_images_per_prompt: Optional[int] = 1, | |
| generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None, | |
| latents: Optional[torch.FloatTensor] = None, | |
| prompt_embeds: Optional[torch.FloatTensor] = None, | |
| pooled_prompt_embeds: Optional[torch.FloatTensor] = None, | |
| output_type: Optional[str] = "pil", | |
| return_dict: bool = True, | |
| joint_attention_kwargs: Optional[Dict[str, Any]] = None, | |
| max_sequence_length: int = 512, | |
| good_vae: Optional[Any] = None, | |
| ): | |
| height = height or self.default_sample_size * self.vae_scale_factor | |
| width = width or self.default_sample_size * self.vae_scale_factor | |
| self.check_inputs( | |
| prompt, | |
| prompt_2, | |
| height, | |
| width, | |
| prompt_embeds=prompt_embeds, | |
| pooled_prompt_embeds=pooled_prompt_embeds, | |
| max_sequence_length=max_sequence_length, | |
| ) | |
| self._guidance_scale = guidance_scale | |
| self._joint_attention_kwargs = joint_attention_kwargs | |
| self._interrupt = False | |
| batch_size = 1 if isinstance(prompt, str) else len(prompt) | |
| device = self._execution_device | |
| lora_scale = joint_attention_kwargs.get("scale", None) if joint_attention_kwargs is not None else None | |
| prompt_embeds, pooled_prompt_embeds, text_ids = self.encode_prompt( | |
| prompt=prompt, | |
| prompt_2=prompt_2, | |
| prompt_embeds=prompt_embeds, | |
| pooled_prompt_embeds=pooled_prompt_embeds, | |
| device=device, | |
| num_images_per_prompt=num_images_per_prompt, | |
| max_sequence_length=max_sequence_length, | |
| lora_scale=lora_scale, | |
| ) | |
| num_channels_latents = self.transformer.config.in_channels // 4 | |
| latents, latent_image_ids = self.prepare_latents( | |
| batch_size * num_images_per_prompt, | |
| num_channels_latents, | |
| height, | |
| width, | |
| prompt_embeds.dtype, | |
| device, | |
| generator, | |
| latents, | |
| ) | |
| sigmas = np.linspace(1.0, 1 / num_inference_steps, num_inference_steps) | |
| image_seq_len = latents.shape[1] | |
| mu = calculate_shift( | |
| image_seq_len, | |
| self.scheduler.config.base_image_seq_len, | |
| self.scheduler.config.max_image_seq_len, | |
| self.scheduler.config.base_shift, | |
| self.scheduler.config.max_shift, | |
| ) | |
| timesteps, num_inference_steps = retrieve_timesteps( | |
| self.scheduler, | |
| num_inference_steps, | |
| device, | |
| timesteps, | |
| sigmas, | |
| mu=mu, | |
| ) | |
| self._num_timesteps = len(timesteps) | |
| guidance = torch.full([1], guidance_scale, device=device, dtype=torch.float32).expand(latents.shape[0]) if self.transformer.config.guidance_embeds else None | |
| for i, t in enumerate(timesteps): | |
| if self.interrupt: | |
| continue | |
| timestep = t.expand(latents.shape[0]).to(latents.dtype) | |
| noise_pred = self.transformer( | |
| hidden_states=latents, | |
| timestep=timestep / 1000, | |
| guidance=guidance, | |
| pooled_projections=pooled_prompt_embeds, | |
| encoder_hidden_states=prompt_embeds, | |
| txt_ids=text_ids, | |
| img_ids=latent_image_ids, | |
| joint_attention_kwargs=self.joint_attention_kwargs, | |
| return_dict=False, | |
| )[0] | |
| latents_for_image = self._unpack_latents(latents, height, width, self.vae_scale_factor) | |
| latents_for_image = (latents_for_image / self.vae.config.scaling_factor) + self.vae.config.shift_factor | |
| image = self.vae.decode(latents_for_image, return_dict=False)[0] | |
| yield self.image_processor.postprocess(image, output_type=output_type)[0] | |
| latents = self.scheduler.step(noise_pred, t, latents, return_dict=False)[0] | |
| torch.cuda.empty_cache() | |
| latents = self._unpack_latents(latents, height, width, self.vae_scale_factor) | |
| latents = (latents / good_vae.config.scaling_factor) + good_vae.config.shift_factor | |
| image = good_vae.decode(latents, return_dict=False)[0] | |
| self.maybe_free_model_hooks() | |
| torch.cuda.empty_cache() | |
| yield self.image_processor.postprocess(image, output_type=output_type)[0] | |
| #-----------------------------------------------------------------------------------LoRA's--------------------------------------------------------------------------# | |
| loras = [ | |
| #0 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Castor-Dramatic-Neon-Flux-LoRA/resolve/main/images/DN2.webp", | |
| "title": "Dramatic Neon", | |
| "repo": "prithivMLmods/Castor-Dramatic-Neon-Flux-LoRA", | |
| "weights": "Dramatic-Neon-Flux-LoRA.safetensors", | |
| "trigger_word": "Dramatic Neon" | |
| }, | |
| #1 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Canopus-LoRA-Flux-FaceRealism/resolve/main/images/11.png", | |
| "title": "Flux Face Realism", | |
| "repo": "prithivMLmods/Canopus-LoRA-Flux-FaceRealism", | |
| "trigger_word": "Realism" | |
| }, | |
| #2 | |
| { | |
| "image": "https://huggingface.co/alvdansen/softserve_anime/resolve/main/images/ComfyUI_00134_.png", | |
| "title": "Softserve Anime", | |
| "repo": "alvdansen/softserve_anime", | |
| "trigger_word": "sftsrv style illustration" | |
| }, | |
| #3 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Canopus-LoRA-Flux-Anime/resolve/main/assets/4.png", | |
| "title": "Flux Anime", | |
| "repo": "prithivMLmods/Canopus-LoRA-Flux-Anime", | |
| "trigger_word": "Anime" | |
| }, | |
| #4 | |
| { | |
| "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-One-Click-Creative-Template/resolve/main/images/f2cc649985648e57b9b9b14ca7a8744ac8e50d75b3a334ed4df0f368.jpg", | |
| "title": "Creative Template", | |
| "repo": "Shakker-Labs/FLUX.1-dev-LoRA-One-Click-Creative-Template", | |
| "trigger_word": "The background is 4 real photos, and in the middle is a cartoon picture summarizing the real photos." | |
| }, | |
| #5 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0/resolve/main/images/3.png", | |
| "title": "Ultra Realism", | |
| "repo": "prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0", | |
| "trigger_word": "Ultra realistic" | |
| }, | |
| #6 | |
| { | |
| "image": "https://huggingface.co/gokaygokay/Flux-Game-Assets-LoRA-v2/resolve/main/images/example_y2bqpuphc.png", | |
| "title": "Game Assets", | |
| "repo": "gokaygokay/Flux-Game-Assets-LoRA-v2", | |
| "trigger_word": "wbgmsst, white background" | |
| }, | |
| #7 | |
| { | |
| "image": "https://huggingface.co/alvdansen/softpasty-flux-dev/resolve/main/images/ComfyUI_00814_%20(2).png", | |
| "title": "Softpasty", | |
| "repo": "alvdansen/softpasty-flux-dev", | |
| "trigger_word": "araminta_illus illustration style" | |
| }, | |
| #8 | |
| { | |
| "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-add-details/resolve/main/images/0.png", | |
| "title": "Details Add", | |
| "repo": "Shakker-Labs/FLUX.1-dev-LoRA-add-details", | |
| "trigger_word": "" | |
| }, | |
| #9 | |
| { | |
| "image": "https://huggingface.co/alvdansen/frosting_lane_flux/resolve/main/images/content%20-%202024-08-11T010011.238.jpeg", | |
| "title": "Frosting Lane", | |
| "repo": "alvdansen/frosting_lane_flux", | |
| "trigger_word": "frstingln illustration" | |
| }, | |
| #10 | |
| { | |
| "image": "https://huggingface.co/aleksa-codes/flux-ghibsky-illustration/resolve/main/images/example5.jpg", | |
| "title": "Ghibsky Illustration", | |
| "repo": "aleksa-codes/flux-ghibsky-illustration", | |
| "trigger_word": "GHIBSKY style painting" | |
| }, | |
| #11 | |
| { | |
| "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-Dark-Fantasy/resolve/main/images/c2215bd73da9f14fcd63cc93350e66e2901bdafa6fb8abaaa2c32a1b.jpg", | |
| "title": "Dark Fantasy", | |
| "repo": "Shakker-Labs/FLUX.1-dev-LoRA-Dark-Fantasy", | |
| "trigger_word": "" | |
| }, | |
| #12 | |
| { | |
| "image": "https://huggingface.co/Norod78/Flux_1_Dev_LoRA_Paper-Cutout-Style/resolve/main/d13591878d5043f3989dd6eb1c25b710_233c18effb4b491cb467ca31c97e90b5.png", | |
| "title": "Paper Cutout", | |
| "repo": "Norod78/Flux_1_Dev_LoRA_Paper-Cutout-Style", | |
| "trigger_word": "Paper Cutout Style" | |
| }, | |
| #13 | |
| { | |
| "image": "https://huggingface.co/alvdansen/mooniverse/resolve/main/images/out-0%20(17).webp", | |
| "title": "Mooniverse", | |
| "repo": "alvdansen/mooniverse", | |
| "trigger_word": "surreal style" | |
| }, | |
| #14 | |
| { | |
| "image": "https://huggingface.co/alvdansen/pola-photo-flux/resolve/main/images/out-0%20-%202024-09-22T130819.351.webp", | |
| "title": "Pola Photo", | |
| "repo": "alvdansen/pola-photo-flux", | |
| "trigger_word": "polaroid style" | |
| }, | |
| #15 | |
| { | |
| "image": "https://huggingface.co/multimodalart/flux-tarot-v1/resolve/main/images/7e180627edd846e899b6cd307339140d_5b2a09f0842c476b83b6bd2cb9143a52.png", | |
| "title": "Flux Tarot", | |
| "repo": "multimodalart/flux-tarot-v1", | |
| "trigger_word": "in the style of TOK a trtcrd tarot style" | |
| }, | |
| #16 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Flux-Dev-Real-Anime-LoRA/resolve/main/images/111.png", | |
| "title": "Real Anime", | |
| "repo": "prithivMLmods/Flux-Dev-Real-Anime-LoRA", | |
| "trigger_word": "Real Anime" | |
| }, | |
| #17 | |
| { | |
| "image": "https://huggingface.co/diabolic6045/Flux_Sticker_Lora/resolve/main/images/example_s3pxsewcb.png", | |
| "title": "Stickers", | |
| "repo": "diabolic6045/Flux_Sticker_Lora", | |
| "trigger_word": "5t1cker 5ty1e" | |
| }, | |
| #18 | |
| { | |
| "image": "https://huggingface.co/VideoAditor/Flux-Lora-Realism/resolve/main/images/feel-the-difference-between-using-flux-with-lora-from-xlab-v0-j0ehybmvxehd1.png", | |
| "title": "Realism", | |
| "repo": "XLabs-AI/flux-RealismLora", | |
| "trigger_word": "" | |
| }, | |
| #19 | |
| { | |
| "image": "https://huggingface.co/alvdansen/flux-koda/resolve/main/images/ComfyUI_00583_%20(1).png", | |
| "title": "Koda", | |
| "repo": "alvdansen/flux-koda", | |
| "trigger_word": "flmft style" | |
| }, | |
| #20 | |
| { | |
| "image": "https://huggingface.co/mgwr/Cine-Aesthetic/resolve/main/images/00019-1333633802.png", | |
| "title": "Cine Aesthetic", | |
| "repo": "mgwr/Cine-Aesthetic", | |
| "trigger_word": "mgwr/cine" | |
| }, | |
| #21 | |
| { | |
| "image": "https://huggingface.co/SebastianBodza/flux_cute3D/resolve/main/images/astronaut.webp", | |
| "title": "Cute 3D", | |
| "repo": "SebastianBodza/flux_cute3D", | |
| "trigger_word": "NEOCUTE3D" | |
| }, | |
| #22 | |
| { | |
| "image": "https://huggingface.co/bingbangboom/flux_dreamscape/resolve/main/images/3.jpg", | |
| "title": "Dreamscape", | |
| "repo": "bingbangboom/flux_dreamscape", | |
| "trigger_word": "in the style of BSstyle004" | |
| }, | |
| #23 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Canopus-LoRA-Flux-FaceRealism/resolve/main/images/xc.webp", | |
| "title": "Cute Kawaii", | |
| "repo": "prithivMLmods/Canopus-Cute-Kawaii-Flux-LoRA", | |
| "trigger_word": "cute-kawaii" | |
| }, | |
| #24 | |
| { | |
| "image": "https://cdn-uploads.huggingface.co/production/uploads/64b24543eec33e27dc9a6eca/_jyra-jKP_prXhzxYkg1O.png", | |
| "title": "Pastel Anime", | |
| "repo": "Raelina/Flux-Pastel-Anime", | |
| "trigger_word": "Anime" | |
| }, | |
| #25 | |
| { | |
| "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-Vector-Journey/resolve/main/images/f7a66b51c89896854f31bef743dc30f33c6ea3c0ed8f9ff04d24b702.jpg", | |
| "title": "Vector", | |
| "repo": "Shakker-Labs/FLUX.1-dev-LoRA-Vector-Journey", | |
| "trigger_word": "artistic style blends reality and illustration elements" | |
| }, | |
| #26 | |
| { | |
| "image": "https://huggingface.co/bingbangboom/flux-miniature-worlds/resolve/main/images/2.jpg", | |
| "title": "Miniature", | |
| "repo": "bingbangboom/flux-miniature-worlds", | |
| "weights": "flux_MNTRWRLDS.safetensors", | |
| "trigger_word": "Image in the style of MNTRWRLDS" | |
| }, | |
| #27 | |
| { | |
| "image": "https://huggingface.co/glif-loradex-trainer/bingbangboom_flux_surf/resolve/main/samples/1729012111574__000002000_0.jpg", | |
| "title": "Surf Bingbangboom", | |
| "repo": "glif-loradex-trainer/bingbangboom_flux_surf", | |
| "weights": "flux_surf.safetensors", | |
| "trigger_word": "SRFNGV01" | |
| }, | |
| #28 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Canopus-Snoopy-Charlie-Brown-Flux-LoRA/resolve/main/000.png", | |
| "title": "Snoopy Charlie", | |
| "repo": "prithivMLmods/Canopus-Snoopy-Charlie-Brown-Flux-LoRA", | |
| "trigger_word": "Snoopy Charlie Brown" | |
| }, | |
| #29 | |
| { | |
| "image": "https://huggingface.co/alvdansen/sonny-anime-fixed/resolve/main/images/uqAuIMqA6Z7mvPkHg4qJE_f4c3cbe64e0349e7b946d02adeacdca3.png", | |
| "title": "Fixed Sonny", | |
| "repo": "alvdansen/sonny-anime-fixed", | |
| "trigger_word": "nm22 style" | |
| }, | |
| #30 | |
| { | |
| "image": "https://huggingface.co/davisbro/flux-multi-angle/resolve/main/multi-angle-examples/3.png", | |
| "title": "Multi Angle", | |
| "repo": "davisbro/flux-multi-angle", | |
| "trigger_word": "A TOK composite photo of a person posing at different angles" | |
| }, | |
| #31 | |
| { | |
| "image": "https://huggingface.co/glif/how2draw/resolve/main/images/glif-how2draw-araminta-k-vbnvy94npt8m338r2vm02m50.jpg", | |
| "title": "How2Draw", | |
| "repo": "glif/how2draw", | |
| "trigger_word": "How2Draw" | |
| }, | |
| #32 | |
| { | |
| "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-Text-Poster/resolve/main/images/6dd1a918d89991ad5e40513ab88e7d892077f89dac93edcf4b660dd2.jpg", | |
| "title": "Text Poster", | |
| "repo": "Shakker-Labs/FLUX.1-dev-LoRA-Text-Poster", | |
| "trigger_word": "text poster" | |
| }, | |
| #33 | |
| { | |
| "image": "https://huggingface.co/SebastianBodza/Flux_Aquarell_Watercolor_v2/resolve/main/images/coffee.webp", | |
| "title": "Aquarell Watercolor", | |
| "repo": "SebastianBodza/Flux_Aquarell_Watercolor_v2", | |
| "trigger_word": "AQUACOLTOK" | |
| }, | |
| #34 | |
| { | |
| "image": "https://huggingface.co/Purz/face-projection/resolve/main/34031797.jpeg", | |
| "title": "Face Projection ", | |
| "repo": "Purz/face-projection", | |
| "trigger_word": "f4c3_p40j3ct10n" | |
| }, | |
| #35 | |
| { | |
| "image": "https://huggingface.co/martintomov/ecom-flux-v2/resolve/main/images/example_z30slf97z.png", | |
| "title": "Ecom Design Art", | |
| "repo": "martintomov/ecom-flux-v2", | |
| "trigger_word": "" | |
| }, | |
| #36 | |
| { | |
| "image": "https://huggingface.co/TheAwakenOne/max-headroom/resolve/main/sample/max-headroom_000900_00_20241015234926.png", | |
| "title": "Max Head-Room", | |
| "repo": "TheAwakenOne/max-headroom", | |
| "weights": "max-headroom-v1.safetensors", | |
| "trigger_word": "M2X, Max-Headroom" | |
| }, | |
| #37 | |
| { | |
| "image": "https://huggingface.co/renderartist/toyboxflux/resolve/main/images/3D__00366_.png", | |
| "title": "Toy Box Flux", | |
| "repo": "renderartist/toyboxflux", | |
| "weights": "Toy_Box_Flux_v2_renderartist.safetensors", | |
| "trigger_word": "t0yb0x, simple toy design, detailed toy design, 3D render" | |
| }, | |
| #38 | |
| { | |
| "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-live-3D/resolve/main/images/51a716fb6fe9ba5d54c260b70e7ff661d38acedc7fb725552fa77bcf.jpg", | |
| "title": "Live 3D", | |
| "repo": "Shakker-Labs/FLUX.1-dev-LoRA-live-3D", | |
| "trigger_word": "" | |
| }, | |
| #39 | |
| { | |
| "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-Garbage-Bag-Art/resolve/main/images/42e944819b43869a03dc252d10409b5944a62494c7082816121016f9.jpg", | |
| "title": "Garbage Bag Art", | |
| "repo": "Shakker-Labs/FLUX.1-dev-LoRA-Garbage-Bag-Art", | |
| "trigger_word": "Inflatable plastic bag" | |
| }, | |
| #40 | |
| { | |
| "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-Logo-Design/resolve/main/images/73e7db6a33550d05836ce285549de60075d05373c7b0660d631dac33.jpg", | |
| "title": "Logo Design", | |
| "repo": "Shakker-Labs/FLUX.1-dev-LoRA-Logo-Design", | |
| "trigger_word": "wablogo, logo, Minimalist" | |
| }, | |
| #41 | |
| { | |
| "image": "https://huggingface.co/punzel/flux_sadie_sink/resolve/main/images/ComfyUI_Flux_Finetune_00069_.png", | |
| "title": "Sadie Sink", | |
| "repo": "punzel/flux_sadie_sink", | |
| "weights": "flux_sadie_sink.safetensors", | |
| "trigger_word": "Sadie Sink" | |
| }, | |
| #42 | |
| { | |
| "image": "https://huggingface.co/punzel/flux_jenna_ortega/resolve/main/images/ComfyUI_Flux_Finetune_00065_.png", | |
| "title": "Jenna ortega", | |
| "repo": "punzel/flux_jenna_ortega", | |
| "weights": "flux_jenna_ortega.safetensors", | |
| "trigger_word": "Jenna ortega" | |
| }, | |
| #43 | |
| { | |
| "image": "https://huggingface.co/Wakkamaruh/balatro-poker-cards/resolve/main/samples/01.png", | |
| "title": "Poker Cards", | |
| "repo": "Wakkamaruh/balatro-poker-cards", | |
| "weights": "balatro-poker-cards.safetensors", | |
| "trigger_word": "balatrocard" | |
| }, | |
| #44 | |
| { | |
| "image": "https://huggingface.co/lichorosario/flux-cubist-cartoon/resolve/main/samples/albert-einstein.png", | |
| "title": "Cubist Cartoon", | |
| "repo": "lichorosario/flux-cubist-cartoon", | |
| "weights": "lora.safetensors", | |
| "trigger_word": "CBSTCRTN" | |
| }, | |
| #45 | |
| { | |
| "image": "https://huggingface.co/iliketoasters/miniature-people/resolve/main/images/1757-over%20the%20shoulder%20shot%2C%20raw%20photo%2C%20a%20min-fluxcomfy-orgflux1-dev-fp8-128443497-converted.png", | |
| "title": "Miniature People", | |
| "repo": "iliketoasters/miniature-people", | |
| "trigger_word": "miniature people" | |
| }, | |
| #46 | |
| { | |
| "image": "https://huggingface.co/ampp/rough-kids-illustrations/resolve/main/samples/1725115106736__000001000_0.jpg", | |
| "title": "kids Illustrations", | |
| "repo": "ampp/rough-kids-illustrations", | |
| "weights": "rough-kids-illustrations.safetensors", | |
| "trigger_word": "r0ughkids4rt" | |
| }, | |
| #47 | |
| { | |
| "image": "https://huggingface.co/lichorosario/flux-lora-tstvctr/resolve/main/images/example_mo3jx93o6.png", | |
| "title": "TSTVCTR Cartoon", | |
| "repo": "lichorosario/flux-lora-tstvctr", | |
| "weights": "lora.safetensors", | |
| "trigger_word": "TSTVCTR cartoon illustration" | |
| }, | |
| #48 | |
| { | |
| "image": "https://huggingface.co/lichorosario/flux-lora-gliff-tosti-vector-no-captions-2500s/resolve/main/images/example_i6h6fi9sq.png", | |
| "title": "Tosti Vector", | |
| "repo": "lichorosario/flux-lora-gliff-tosti-vector-no-captions-2500s", | |
| "weights": "flux_dev_tosti_vector_without_captions_000002500.safetensors", | |
| "trigger_word": "" | |
| }, | |
| #49 | |
| { | |
| "image": "https://huggingface.co/AlekseyCalvin/Propaganda_Poster_Schnell_by_doctor_diffusion/resolve/main/Trashy.png", | |
| "title": "Propaganda Poster", | |
| "repo": "AlekseyCalvin/Propaganda_Poster_Schnell_by_doctor_diffusion", | |
| "weights": "propaganda_schnell_v1.safetensors", | |
| "trigger_word": "propaganda poster" | |
| }, | |
| #50 | |
| { | |
| "image": "https://huggingface.co/WizWhite/Wiz-PunchOut_Ringside_Portrait/resolve/main/images/punch0ut__ringside_pixel_portrait_depicting_chris_brown_wearing_a_veil__moonstone_gray_background_with_white_ropes___1923906484.png", | |
| "title": "Ringside Portrait", | |
| "repo": "WizWhite/Wiz-PunchOut_Ringside_Portrait", | |
| "trigger_word": "punch0ut, ringside pixel portrait depicting" | |
| }, | |
| #51 | |
| { | |
| "image": "https://huggingface.co/glif-loradex-trainer/kklors_flux_dev_long_exposure/resolve/main/samples/1729016926778__000003000_3.jpg", | |
| "title": "Long Exposure", | |
| "repo": "glif-loradex-trainer/kklors_flux_dev_long_exposure", | |
| "weights": "flux_dev_long_exposure.safetensors", | |
| "trigger_word": "LE" | |
| }, | |
| #52 | |
| { | |
| "image": "https://huggingface.co/DamarJati/streetwear-flux/resolve/main/img/79e891f9-ceb8-4f8a-a51d-bb432789d037.jpeg", | |
| "title": "Street Wear", | |
| "repo": "DamarJati/streetwear-flux", | |
| "weights": "Streetwear.safetensors", | |
| "trigger_word": "Handling Information Tshirt template" | |
| }, | |
| #53 | |
| { | |
| "image": "https://huggingface.co/multimodalart/vintage-ads-flux/resolve/main/samples/-FMpgla6rQ1hBwBpbr-Ao_da7b23c29de14a9cad94901879ae2e2b.png", | |
| "title": "Vintage Ads Flux", | |
| "repo": "multimodalart/vintage-ads-flux", | |
| "weights": "vintage-ads-flux-1350.safetensors", | |
| "trigger_word": "a vintage ad of" | |
| }, | |
| #54 | |
| { | |
| "image": "https://huggingface.co/multimodalart/product-design/resolve/main/images/example_vgv87rlfl.png", | |
| "title": "Product Design", | |
| "repo": "multimodalart/product-design", | |
| "weights": "product-design.safetensors", | |
| "trigger_word": "product designed by prdsgn" | |
| }, | |
| #55 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Canopus-LoRA-Flux-Typography-ASCII/resolve/main/images/NNN.png", | |
| "title": "Typography", | |
| "repo": "prithivMLmods/Canopus-LoRA-Flux-Typography-ASCII", | |
| "weights": "Typography.safetensors", | |
| "trigger_word": "Typography, ASCII Art" | |
| }, | |
| #56 | |
| { | |
| "image": "https://huggingface.co/mateo-19182/mosoco/resolve/main/samples/1725714834007__000002000_0.jpg", | |
| "title": "Mosoco", | |
| "repo": "mateo-19182/mosoco", | |
| "weights": "mosoco.safetensors", | |
| "trigger_word": "moscos0" | |
| }, | |
| #57 | |
| { | |
| "image": "https://huggingface.co/jakedahn/flux-latentpop/resolve/main/images/2.webp", | |
| "title": "Latent Pop", | |
| "repo": "jakedahn/flux-latentpop", | |
| "weights": "lora.safetensors", | |
| "trigger_word": "latentpop" | |
| }, | |
| #58 | |
| { | |
| "image": "https://huggingface.co/glif-loradex-trainer/ddickinson_dstyl3xl/resolve/main/samples/1728556571974__000001500_2.jpg", | |
| "title": "Dstyl3xl", | |
| "repo": "glif-loradex-trainer/ddickinson_dstyl3xl", | |
| "weights": "dstyl3xl.safetensors", | |
| "trigger_word": "in the style of dstyl3xl" | |
| }, | |
| #59 | |
| { | |
| "image": "https://huggingface.co/TDN-M/RetouchFLux/resolve/main/images/496f0680-0158-4f37-805d-d227c1a08a7b.png", | |
| "title": "Retouch FLux", | |
| "repo": "TDN-M/RetouchFLux", | |
| "weights": "TDNM_Retouch.safetensors", | |
| "trigger_word": "luxury, enhance, hdr" | |
| }, | |
| #60 | |
| { | |
| "image": "https://huggingface.co/glif/anime-blockprint-style/resolve/main/images/glif-block-print-anime-flux-dev-araminta-k-lora-araminta-k-e35k8xqsrb8dtq2qcv4gsr3z.jpg", | |
| "title": "Block Print", | |
| "repo": "glif/anime-blockprint-style", | |
| "weights": "bwmanga.safetensors", | |
| "trigger_word": "blockprint style" | |
| }, | |
| #61 | |
| { | |
| "image": "https://huggingface.co/renderartist/weirdthingsflux/resolve/main/images/3D__02303_.png", | |
| "title": "Weird Things Flux", | |
| "repo": "renderartist/weirdthingsflux", | |
| "weights": "Weird_Things_Flux_v1_renderartist.safetensors", | |
| "trigger_word": "w3irdth1ngs, illustration" | |
| }, | |
| #62 | |
| { | |
| "image": "https://replicate.delivery/yhqm/z7f2OBcvga07dCoJ4FeRGZCbE5PvipLhogPhEeU7BazIg5lmA/out-0.webp", | |
| "title": "Replicate Flux LoRA", | |
| "repo": "lucataco/ReplicateFluxLoRA", | |
| "weights": "flux_train_replicate.safetensors", | |
| "trigger_word": "TOK" | |
| }, | |
| #63 | |
| { | |
| "image": "https://huggingface.co/alvdansen/haunted_linework_flux/resolve/main/images/ComfyUI_00755_.png", | |
| "title": "Linework", | |
| "repo": "alvdansen/haunted_linework_flux", | |
| "weights": "hauntedlinework_flux_araminta_k.safetensors", | |
| "trigger_word": "hntdlnwrk style" | |
| }, | |
| #64 | |
| { | |
| "image": "https://huggingface.co/fofr/flux-cassette-futurism/resolve/main/images/example_qgry9jnkj.png", | |
| "title": "Cassette Futurism", | |
| "repo": "fofr/flux-cassette-futurism", | |
| "weights": "lora.safetensors", | |
| "trigger_word": "cassette futurism" | |
| }, | |
| #65 | |
| { | |
| "image": "https://huggingface.co/Wadaka/Mojo_Style_LoRA/resolve/main/Samples/Sample2.png", | |
| "title": "Mojo Style", | |
| "repo": "Wadaka/Mojo_Style_LoRA", | |
| "weights": "Mojo_Style_LoRA.safetensors", | |
| "trigger_word": "Mojo_Style" | |
| }, | |
| #66 | |
| { | |
| "image": "https://huggingface.co/Norod78/JojosoStyle-flux-lora/resolve/main/samples/1725244218477__000004255_1.jpg", | |
| "title": "Jojoso Style", | |
| "repo": "Norod78/JojosoStyle-flux-lora", | |
| "weights": "JojosoStyle_flux_lora.safetensors", | |
| "trigger_word": "JojosoStyle" | |
| }, | |
| #67 | |
| { | |
| "image": "https://huggingface.co/Chunte/flux-lora-Huggieverse/resolve/main/images/Happy%20star.png", | |
| "title": "Huggieverse", | |
| "repo": "Chunte/flux-lora-Huggieverse", | |
| "weights": "lora.safetensors", | |
| "trigger_word": "HGGRE" | |
| }, | |
| #68 | |
| { | |
| "image": "https://huggingface.co/diabolic6045/Flux_Wallpaper_Lora/resolve/main/images/example_hjp51et93.png", | |
| "title": "Wallpaper LoRA", | |
| "repo": "diabolic6045/Flux_Wallpaper_Lora", | |
| "weights": "tost-2024-09-20-07-35-44-wallpap3r5.safetensors", | |
| "trigger_word": "wallpap3r5" | |
| }, | |
| #69 | |
| { | |
| "image": "https://huggingface.co/bingbangboom/flux_geopop/resolve/main/extras/5.png", | |
| "title": "Geo Pop", | |
| "repo": "bingbangboom/flux_geopop", | |
| "weights": "geopop_NWGMTRCPOPV01.safetensors", | |
| "trigger_word": "illustration in the style of NWGMTRCPOPV01" | |
| }, | |
| #70 | |
| { | |
| "image": "https://huggingface.co/bingbangboom/flux_colorscape/resolve/main/images/4.jpg", | |
| "title": "Colorscape", | |
| "repo": "bingbangboom/flux_colorscape", | |
| "weights": "flux_colorscape.safetensors", | |
| "trigger_word": "illustration in the style of ASstyle001" | |
| }, | |
| #71 | |
| { | |
| "image": "https://huggingface.co/dvyio/flux-lora-thermal-image/resolve/main/images/WROSaNNU4-Gw0r5QoBRjf_f164ffa4f0804e68bad1d06d30deecfa.jpg", | |
| "title": "Thermal Image", | |
| "repo": "dvyio/flux-lora-thermal-image", | |
| "weights": "79b5004c57ef4c4390dead1c65977bbb_pytorch_lora_weights.safetensors", | |
| "trigger_word": "thermal image in the style of THRML" | |
| }, | |
| #72 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Canopus-Clothing-Flux-LoRA/resolve/main/images/333.png", | |
| "title": "Clothing Flux", | |
| "repo": "prithivMLmods/Canopus-Clothing-Flux-LoRA", | |
| "weights": "Canopus-Clothing-Flux-Dev-Florence2-LoRA.safetensors", | |
| "trigger_word": "Hoodie, Clothes, Shirt, Pant" | |
| }, | |
| #73 | |
| { | |
| "image": "https://huggingface.co/dvyio/flux-lora-stippled-illustration/resolve/main/images/57FPpbu74QTV45w6oNOtZ_26832270585f456c99e4a98b1c073745.jpg", | |
| "title": "Stippled Illustration", | |
| "repo": "dvyio/flux-lora-stippled-illustration", | |
| "weights": "31984be602a04a1fa296d9ccb244fb29_pytorch_lora_weights.safetensors", | |
| "trigger_word": "stippled illustration in the style of STPPLD" | |
| }, | |
| #74 | |
| { | |
| "image": "https://huggingface.co/wayned/fruitlabels/resolve/main/images/ComfyUI_03969_.png", | |
| "title": "Fruitlabels", | |
| "repo": "wayned/fruitlabels", | |
| "weights": "fruitlabels2.safetensors", | |
| "trigger_word": "fruit labels" | |
| }, | |
| #75 | |
| { | |
| "image": "https://huggingface.co/punzel/flux_margot_robbie/resolve/main/images/ComfyUI_Flux_Finetune_00142_.png", | |
| "title": "Margot Robbie", | |
| "repo": "punzel/flux_margot_robbie", | |
| "weights": "flux_margot_robbie.safetensors", | |
| "trigger_word": "" | |
| }, | |
| #76 | |
| { | |
| "image": "https://huggingface.co/diabolic6045/Formula1_Lego_Lora/resolve/main/images/example_502kcuiba.png", | |
| "title": "Formula 1 Lego", | |
| "repo": "punzel/flux_margot_robbie", | |
| "weights": "tost-2024-09-20-09-58-33-f1leg0s.safetensors", | |
| "trigger_word": "f1leg0s" | |
| }, | |
| #77 | |
| { | |
| "image": "https://huggingface.co/glif/Brain-Melt-Acid-Art/resolve/main/images/IMG_0832.png", | |
| "title": "Melt Acid", | |
| "repo": "glif/Brain-Melt-Acid-Art", | |
| "weights": "Brain_Melt.safetensors", | |
| "trigger_word": "in an acid surrealism style, maximalism" | |
| }, | |
| #78 | |
| { | |
| "image": "https://huggingface.co/jeremytai/enso-zen/resolve/main/images/example_a0iwdj5lu.png", | |
| "title": "Enso", | |
| "repo": "jeremytai/enso-zen", | |
| "weights": "enso-zen.safetensors", | |
| "trigger_word": "enso" | |
| }, | |
| #79 | |
| { | |
| "image": "https://huggingface.co/veryVANYA/opus-ascii-flux/resolve/main/31654332.jpeg", | |
| "title": "Opus Ascii", | |
| "repo": "veryVANYA/opus-ascii-flux", | |
| "weights": "flux_opus_ascii.safetensors", | |
| "trigger_word": "opus_ascii" | |
| }, | |
| #80 | |
| { | |
| "image": "https://huggingface.co/crystantine/cybrpnkz/resolve/main/images/example_plyxk0lej.png", | |
| "title": "Cybrpnkz", | |
| "repo": "crystantine/cybrpnkz", | |
| "weights": "cybrpnkz.safetensors", | |
| "trigger_word": "architecture style of CYBRPNKZ" | |
| }, | |
| #81 | |
| { | |
| "image": "https://huggingface.co/fyp1/pattern_generation/resolve/main/images/1727560066052__000001000_7.jpg", | |
| "title": "Pattern Generation", | |
| "repo": "fyp1/pattern_generation", | |
| "weights": "flux_dev_finetune.safetensors", | |
| "trigger_word": "pattern" | |
| }, | |
| #82 | |
| { | |
| "image": "https://huggingface.co/TheAwakenOne/caricature/resolve/main/sample/caricature_000900_03_20241007143412.png", | |
| "title": "Caricature", | |
| "repo": "TheAwakenOne/caricature", | |
| "weights": "caricature.safetensors", | |
| "trigger_word": "CCTUR3" | |
| }, | |
| #83 | |
| { | |
| "image": "https://huggingface.co/davidrd123/Flux-MoonLanding76-Replicate/resolve/main/images/example_6adktoq5m.png", | |
| "title": "MoonLanding 76", | |
| "repo": "davidrd123/Flux-MoonLanding76-Replicate", | |
| "weights": "lora.safetensors", | |
| "trigger_word": "m00nl4nd1ng" | |
| }, | |
| #84 | |
| { | |
| "image": "https://huggingface.co/Purz/neon-sign/resolve/main/33944768.jpeg", | |
| "title": "Neon", | |
| "repo": "Purz/neon-sign", | |
| "weights": "purz-n30n_51gn.safetensors", | |
| "trigger_word": "n30n_51gn" | |
| }, | |
| #85 | |
| { | |
| "image": "https://huggingface.co/WizWhite/wizard-s-vintage-sardine-tins/resolve/main/27597694.jpeg", | |
| "title": "Vintage Sardine Tins", | |
| "repo": "WizWhite/wizard-s-vintage-sardine-tins", | |
| "weights": "Wiz-SardineTins_Flux.safetensors", | |
| "trigger_word": "Vintage Sardine Tin, Tinned Fish, vintage xyz tin" | |
| }, | |
| #86 | |
| { | |
| "image": "https://huggingface.co/TheAwakenOne/mtdp-balloon-character/resolve/main/sample/mtdp-balloon-character_000200_01_20241014221110.png", | |
| "title": "Float Ballon Character", | |
| "repo": "TheAwakenOne/mtdp-balloon-character", | |
| "weights": "mtdp-balloon-character.safetensors", | |
| "trigger_word": "FLOAT" | |
| }, | |
| #87 | |
| { | |
| "image": "https://huggingface.co/glif/golden-haggadah/resolve/main/images/6aca6403-ecd6-4216-a66a-490ae25ff1b2.jpg", | |
| "title": "Golden Haggadah", | |
| "repo": "glif/golden-haggadah", | |
| "weights": "golden_haggadah.safetensors", | |
| "trigger_word": "golden haggadah style" | |
| }, | |
| #88 | |
| { | |
| "image": "https://huggingface.co/glif-loradex-trainer/usernametaken420__oz_ftw_balaclava/resolve/main/samples/1729278631255__000001500_1.jpg", | |
| "title": "Ftw Balaclava", | |
| "repo": "glif-loradex-trainer/usernametaken420__oz_ftw_balaclava", | |
| "weights": "oz_ftw_balaclava.safetensors", | |
| "trigger_word": "ftw balaclava" | |
| }, | |
| #89 | |
| { | |
| "image": "https://huggingface.co/AlloReview/flux-lora-undraw/resolve/main/images/Flux%20Lora%20Undraw%20Prediction.webp", | |
| "title": "Undraw", | |
| "repo": "AlloReview/flux-lora-undraw", | |
| "weights": "lora.safetensors", | |
| "trigger_word": "in the style of UndrawPurple" | |
| }, | |
| #90 | |
| { | |
| "image": "https://huggingface.co/Disra/lora-anime-test-02/resolve/main/assets/image_0_0.png", | |
| "title": "Anime Test", | |
| "repo": "Disra/lora-anime-test-02", | |
| "weights": "pytorch_lora_weights.safetensors", | |
| "trigger_word": "anime" | |
| }, | |
| #91 | |
| { | |
| "image": "https://huggingface.co/wanghaofan/Black-Myth-Wukong-FLUX-LoRA/resolve/main/images/7d0ac495a4d5e4a3a30df25f08379a3f956ef99e1dc3e252fc1fca3a.jpg", | |
| "title": "Black Myth Wukong", | |
| "repo": "wanghaofan/Black-Myth-Wukong-FLUX-LoRA", | |
| "weights": "pytorch_lora_weights.safetensors", | |
| "trigger_word": "wukong" | |
| }, | |
| #92 | |
| { | |
| "image": "https://huggingface.co/nerijs/pastelcomic-flux/resolve/main/images/4uZ_vaYg-HQnfa5D9gfli_38bf3f95d8b345e5a9bd42d978a15267.png", | |
| "title": "Pastelcomic", | |
| "repo": "nerijs/pastelcomic-flux", | |
| "weights": "pastelcomic_v1.safetensors", | |
| "trigger_word": "" | |
| }, | |
| #93 | |
| { | |
| "image": "https://huggingface.co/RareConcepts/Flux.1-dev-LoKr-Moonman/resolve/main/assets/image_6_0.png", | |
| "title": "Moonman", | |
| "repo": "RareConcepts/Flux.1-dev-LoKr-Moonman", | |
| "weights": "pytorch_lora_weights.safetensors", | |
| "trigger_word": "moonman" | |
| }, | |
| #94 | |
| { | |
| "image": "https://huggingface.co/martintomov/ascii-flux-v1/resolve/main/images/0af53645-ddcc-4803-93c8-f7e43f6fbbd1.jpeg", | |
| "title": "Ascii Flux", | |
| "repo": "martintomov/ascii-flux-v1", | |
| "weights": "ascii-art-v1.safetensors", | |
| "trigger_word": "ASCII art" | |
| }, | |
| #95 | |
| { | |
| "image": "https://huggingface.co/Omarito2412/Stars-Galaxy-Flux/resolve/main/images/25128409.jpeg", | |
| "title": "Ascii Flux", | |
| "repo": "Omarito2412/Stars-Galaxy-Flux", | |
| "weights": "Stars_Galaxy_Flux.safetensors", | |
| "trigger_word": "mlkwglx" | |
| }, | |
| #96 | |
| { | |
| "image": "https://huggingface.co/brushpenbob/flux-pencil-v2/resolve/main/26193927.jpeg", | |
| "title": "Pencil V2", | |
| "repo": "brushpenbob/flux-pencil-v2", | |
| "weights": "Flux_Pencil_v2_r1.safetensors", | |
| "trigger_word": "evang style" | |
| }, | |
| #97 | |
| { | |
| "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-Children-Simple-Sketch/resolve/main/images/1f20519208cef367af2fda8d91ddbba674f39b097389d12ee25b4cb1.jpg", | |
| "title": "Children Simple Sketch", | |
| "repo": "Shakker-Labs/FLUX.1-dev-LoRA-Children-Simple-Sketch", | |
| "weights": "FLUX-dev-lora-children-simple-sketch.safetensors", | |
| "trigger_word": "sketched style" | |
| }, | |
| #98 | |
| { | |
| "image": "https://huggingface.co/victor/contemporarink/resolve/main/images/example_hnqc22urm.png", | |
| "title": "Contemporarink", | |
| "repo": "victor/contemporarink", | |
| "weights": "inky-colors.safetensors", | |
| "trigger_word": "ECACX" | |
| }, | |
| #99 | |
| { | |
| "image": "https://huggingface.co/wavymulder/OverlordStyleFLUX/resolve/main/imgs/ComfyUI_00668_.png", | |
| "title": "OverlordStyle", | |
| "repo": "wavymulder/OverlordStyleFLUX", | |
| "weights": "ovld_style_overlord_wavymulder.safetensors", | |
| "trigger_word": "ovld style anime" | |
| }, | |
| #100 | |
| { | |
| "image": "https://huggingface.co/marceloxp/canny-quest/resolve/main/26676266.jpeg", | |
| "title": "Canny quest", | |
| "repo": "marceloxp/canny-quest", | |
| "weights": "Canny_Quest-000004.safetensors", | |
| "trigger_word": "blonde, silver silk dress, perfectly round sunglasses, pearl necklace" | |
| }, | |
| #101 | |
| { | |
| "image": "https://huggingface.co/busetolunay/building_flux_lora_v1/resolve/main/samples/1725469125185__000001250_2.jpg", | |
| "title": "Building Flux", | |
| "repo": "busetolunay/building_flux_lora_v1", | |
| "weights": "building_flux_lora_v4.safetensors", | |
| "trigger_word": "a0ce" | |
| }, | |
| #102 | |
| { | |
| "image": "https://huggingface.co/Omarito2412/Tinker-Bell-Flux/resolve/main/images/9e9e7eda-3ddf-467a-a7f8-6d8e3ef80cd0.png", | |
| "title": "Tinker Bell Flux", | |
| "repo": "Omarito2412/Tinker-Bell-Flux", | |
| "weights": "TinkerBellV2-FLUX.safetensors", | |
| "trigger_word": "TinkerWaifu, blue eyes, single hair bun" | |
| }, | |
| #103 | |
| { | |
| "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-playful-metropolis/resolve/main/images/3e9265312b3b726c224a955ec9254a0f95c2c8b78ce635929183a075.jpg", | |
| "title": "Playful Metropolis", | |
| "repo": "Shakker-Labs/FLUX.1-dev-LoRA-playful-metropolis", | |
| "weights": "FLUX-dev-lora-playful_metropolis.safetensors", | |
| "trigger_word": "" | |
| }, | |
| #104 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Castor-Character-Polygon-LoRA/resolve/main/images/1000.webp", | |
| "title": "Character Polygon", | |
| "repo": "prithivMLmods/Castor-Character-Polygon-Flux-LoRA", | |
| "weights": "Castor-Character-Polygon-LoRA.safetensors", | |
| "trigger_word": "3D Polygon" | |
| }, | |
| #105 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Castor-Gta6-Theme-Flux-LoRA/resolve/main/images/gta1.webp", | |
| "title": "GTA 6 Theme", | |
| "repo": "prithivMLmods/Castor-Gta6-Theme-Flux-LoRA", | |
| "weights": "Gta6.safetensors", | |
| "trigger_word": "GTA 6 Theme, World of GTA 6" | |
| }, | |
| #106 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Castor-Concept-Gta6-Character-Design/resolve/main/images/L3.webp", | |
| "title": "GTA Character Concept", | |
| "repo": "prithivMLmods/Castor-Flux-Concept-Gta6-Character-Design", | |
| "weights": "Gta6-Concept-Charecter.safetensors", | |
| "trigger_word": "Jason, Lucia, GTA 6" | |
| }, | |
| #107 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Castor-3D-Sketchfab-Flux-LoRA/resolve/main/images/S1.png", | |
| "title": "3D Sketchfab", | |
| "repo": "prithivMLmods/Castor-3D-Sketchfab-Flux-LoRA", | |
| "weights": "Castor-3D-Sketchfab-Flux-LoRA.safetensors", | |
| "trigger_word": "3D Sketchfab" | |
| }, | |
| #108 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Castor-Collage-Dim-Flux-LoRA/resolve/main/images/C1.webp", | |
| "title": "In Image Collage", | |
| "repo": "prithivMLmods/Castor-Collage-Dim-Flux-LoRA", | |
| "weights": "Castor-Collage-Dim-Flux-LoRA.safetensors", | |
| "trigger_word": "collage" | |
| }, | |
| #109 | |
| { | |
| "image": "https://huggingface.co/brushpenbob/flux-midjourney-anime/resolve/main/25439344.jpeg", | |
| "title": "Anime Journey", | |
| "repo": "brushpenbob/flux-midjourney-anime", | |
| "weights": "FLUX_MidJourney_Anime.safetensors", | |
| "trigger_word": "egmid" | |
| }, | |
| #110 | |
| { | |
| "image": "https://huggingface.co/glif-loradex-trainer/maxxd4240_minimalistPastel/resolve/main/samples/1727255690613__000002500_0.jpg", | |
| "title": "Min Pastel", | |
| "repo": "glif-loradex-trainer/maxxd4240_minimalistPastel", | |
| "weights": "minimalistPastel.safetensors", | |
| "trigger_word": "minimalistPastel" | |
| }, | |
| #111 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Castor-Red-Dead-Redemption-2-Flux-LoRA/resolve/main/images/rdr12.webp", | |
| "title": "RDR2", | |
| "repo": "prithivMLmods/Castor-Red-Dead-Redemption-2-Flux-LoRA", | |
| "weights": "Castor-Red-Dead-Redemption-2-Flux-LoRA.safetensors", | |
| "trigger_word": "Red Dead Redemption 2" | |
| }, | |
| #112 | |
| { | |
| "image": "https://huggingface.co/WizWhite/wizard-s-paper-model-universe/resolve/main/35746354.jpeg", | |
| "title": "Paper Model", | |
| "repo": "WizWhite/wizard-s-paper-model-universe", | |
| "weights": "Wiz-Paper_Model_Universe.safetensors", | |
| "trigger_word": "A paper model" | |
| }, | |
| #113 | |
| { | |
| "image": "https://huggingface.co/renderartist/retrocomicflux/resolve/main/images/ComfyUI_temp_ipugi_00040_.png", | |
| "title": "Retrocomic Flux", | |
| "repo": "renderartist/retrocomicflux", | |
| "weights": "Retro_Comic_Flux_v1_renderartist.safetensors", | |
| "trigger_word": "comic book panel" | |
| }, | |
| #114 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Castor-Happy-Halloween-Flux-LoRA/resolve/main/images/hw1.webp", | |
| "title": "Halloween Flux", | |
| "repo": "prithivMLmods/Castor-Happy-Halloween-Flux-LoRA", | |
| "weights": "Castor-Happy-Halloween-Flux-LoRA.safetensors", | |
| "trigger_word": "happy halloween" | |
| }, | |
| #115 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Castor-3D-Portrait-Flux-LoRA/resolve/main/images/1.webp", | |
| "title": "Castor-3D-Portrait", | |
| "repo": "prithivMLmods/Castor-3D-Portrait-Flux-LoRA", | |
| "weights": "Castor-3D-Portrait-Flux-LoRA.safetensors", | |
| "trigger_word": "3D Portrait" | |
| }, | |
| #116 | |
| { | |
| "image": "https://huggingface.co/renderartist/coloringbookflux/resolve/main/images/ComfyUI_09731_.png", | |
| "title": "Coloring book flux", | |
| "repo": "renderartist/coloringbookflux", | |
| "weights": "c0l0ringb00k_Flux_v1_renderartist.safetensors", | |
| "trigger_word": "c0l0ringb00k, coloring book, coloring book page" | |
| }, | |
| #117 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Uncoloured-Polygon-Flux-LoRA/resolve/main/images/1.webp", | |
| "title": "Uncoloured Polygon", | |
| "repo": "prithivMLmods/Uncoloured-Polygon-Flux-LoRA", | |
| "weights": "Uncoloured-3D-Polygon.safetensors", | |
| "trigger_word": "uncoloured polygon" | |
| }, | |
| #118 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Past-Present-Deep-Mix-Flux-LoRA/resolve/main/images/PP3.webp", | |
| "title": "Past Present Mix", | |
| "repo": "prithivMLmods/Past-Present-Deep-Mix-Flux-LoRA", | |
| "weights": "Past-Present-Deep-Mix-Flux-LoRA.safetensors", | |
| "trigger_word": "Mixing Past and Present" | |
| }, | |
| #119 | |
| { | |
| "image": "https://huggingface.co/gokaygokay/Flux-Double-Exposure-LoRA/resolve/main/images/image3.jpg", | |
| "title": "Double Exposure", | |
| "repo": "gokaygokay/Flux-Double-Exposure-LoRA", | |
| "weights": "double_exposure.safetensors", | |
| "trigger_word": "dblxpsr" | |
| }, | |
| #120 | |
| { | |
| "image": "https://huggingface.co/gokaygokay/Flux-Seamless-Texture-LoRA/resolve/main/images/image3.jpg", | |
| "title": "Seamless Texture", | |
| "repo": "gokaygokay/Flux-Seamless-Texture-LoRA", | |
| "weights": "seamless_texture.safetensors", | |
| "trigger_word": "smlstxtr" | |
| }, | |
| #121 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Mockup-Texture-Flux-LoRA/resolve/main/images/MU1.webp", | |
| "title": "Mockup Texture", | |
| "repo": "prithivMLmods/Mockup-Texture-Flux-LoRA", | |
| "weights": "Mockup-Texture.safetensors", | |
| "trigger_word": "Mockup" | |
| }, | |
| #122 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Ton618-Tarot-Cards-Flux-LoRA/resolve/main/images/c2.webp", | |
| "title": "Tarot Cards", | |
| "repo": "prithivMLmods/Ton618-Tarot-Cards-Flux-LoRA", | |
| "weights": "Tarot-card.safetensors", | |
| "trigger_word": "Tarot card" | |
| }, | |
| #123 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Ton618-Amxtoon-Flux-LoRA/resolve/main/images/am1.webp", | |
| "title": "Amxtoon", | |
| "repo": "prithivMLmods/Ton618-Amxtoon-Flux-LoRA", | |
| "weights": "Amxtoon.safetensors", | |
| "trigger_word": "Amxtoon" | |
| }, | |
| #124 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Ton618-Epic-Realism-Flux-LoRA/resolve/main/images/ep3.png", | |
| "title": "Epic Realism", | |
| "repo": "prithivMLmods/Ton618-Epic-Realism-Flux-LoRA", | |
| "weights": "Epic-Realism-Unpruned.safetensors", | |
| "trigger_word": "Epic Realism" | |
| }, | |
| #125 | |
| { | |
| "image": "https://huggingface.co/bingbangboom/flux-mixReality/resolve/main/images/3.jpg", | |
| "title": "Mixed Reality", | |
| "repo": "bingbangboom/flux-mixReality", | |
| "weights": "HLFILSTHLFPHTO_000002500.safetensors", | |
| "trigger_word": "in the style of HLFILSTHLFPHTO" | |
| }, | |
| #126 | |
| { | |
| "image": "https://huggingface.co/sWizad/pokemon-trainer-sprites-pixelart-flux/resolve/main/26578919.jpeg", | |
| "title": "Pixelart", | |
| "repo": "sWizad/pokemon-trainer-sprites-pixelart-flux", | |
| "weights": "pktrainer_F1-v1-0.safetensors", | |
| "trigger_word": "pixel image of, pixel art" | |
| }, | |
| #127 | |
| { | |
| "image": "https://huggingface.co/bingbangboom/flux_colorscape/resolve/main/images/2.jpg", | |
| "title": "Colorscape", | |
| "repo": "bingbangboom/flux_colorscape", | |
| "weights": "flux_colorscape.safetensors", | |
| "trigger_word": "illustration in the style of ASstyle001" | |
| }, | |
| #128 | |
| { | |
| "image": "https://huggingface.co/UmeAiRT/FLUX.1-dev-LoRA-Modern_Pixel_art/resolve/main/images/c363192f-5fa0-4539-8295-b8d9e3e96747.jpeg", | |
| "title": "Modern Pixel art", | |
| "repo": "UmeAiRT/FLUX.1-dev-LoRA-Modern_Pixel_art", | |
| "weights": "ume_modern_pixelart.safetensors", | |
| "trigger_word": "umempart" | |
| }, | |
| #129 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Ton618-Only-Stickers-Flux-LoRA/resolve/main/images/222.png", | |
| "title": "Sticker", | |
| "repo": "prithivMLmods/Ton618-Only-Stickers-Flux-LoRA", | |
| "weights": "only-stickers.safetensors", | |
| "trigger_word": "Only Sticker" | |
| }, | |
| #130 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Ton618-Space-Wallpaper-LoRA/resolve/main/images/222.png", | |
| "title": "Space Wallpaper", | |
| "repo": "prithivMLmods/Ton618-Space-Wallpaper-LoRA", | |
| "weights": "space-wallpaper-xl.safetensor", | |
| "trigger_word": "Space Wallpaper" | |
| }, | |
| #131 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Canopus-Pixar-3D-Flux-LoRA/resolve/main/images/11111.png", | |
| "title": "Pixar 3D", | |
| "repo": "prithivMLmods/Canopus-Pixar-3D-Flux-LoRA", | |
| "weights": "Canopus-Pixar-3D-FluxDev-LoRA.safetensors", | |
| "trigger_word": "Pixar 3D" | |
| }, | |
| #132 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/EBook-Creative-Cover-Flux-LoRA/resolve/main/images/E2.png", | |
| "title": "EBook Cover", | |
| "repo": "prithivMLmods/EBook-Creative-Cover-Flux-LoRA", | |
| "weights": "EBook-Cover.safetensors", | |
| "trigger_word": "EBook Cover" | |
| }, | |
| #133 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Minimal-Futuristic-Flux-LoRA/resolve/main/images/MF3.png", | |
| "title": "Minimal Futuristic", | |
| "repo": "prithivMLmods/Minimal-Futuristic-Flux-LoRA", | |
| "weights": "Minimal-Futuristic.safetensors", | |
| "trigger_word": "Minimal Futuristic" | |
| }, | |
| #134 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Seamless-Pattern-Design-Flux-LoRA/resolve/main/images/SP1.png", | |
| "title": "Seamless Pattern", | |
| "repo": "prithivMLmods/Seamless-Pattern-Design-Flux-LoRA", | |
| "weights": "Seamless-Pattern-Design.safetensors", | |
| "trigger_word": "Seamless Pattern Design" | |
| }, | |
| #135 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Logo-Design-Flux-LoRA/resolve/main/images/LD1.png", | |
| "title": "Logo Design", | |
| "repo": "prithivMLmods/Logo-Design-Flux-LoRA", | |
| "weights": "Logo-design.safetensors", | |
| "trigger_word": "Logo Design" | |
| }, | |
| #136 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Coloring-Book-Flux-LoRA/resolve/main/images/EB1.png", | |
| "title": "Coloring Book", | |
| "repo": "prithivMLmods/Coloring-Book-Flux-LoRA", | |
| "weights": "coloring-book.safetensors", | |
| "trigger_word": "Coloring Book" | |
| }, | |
| #137 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Intense-Red-Flux-LoRA/resolve/main/images/IR1.png", | |
| "title": "Intense Red", | |
| "repo": "prithivMLmods/Intense-Red-Flux-LoRA", | |
| "weights": "Intense-Red.safetensors", | |
| "trigger_word": "Intense Red" | |
| }, | |
| #138 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Glowing-Body-Flux-LoRA/resolve/main/images/GB3.png", | |
| "title": "Glowing Body Flux", | |
| "repo": "prithivMLmods/Glowing-Body-Flux-LoRA", | |
| "weights": "Glowing-Body.safetensors", | |
| "trigger_word": "Glowing Body" | |
| }, | |
| #139 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Electric-Blue-Flux-LoRA/resolve/main/images/EB3.png", | |
| "title": "Electric Blue", | |
| "repo": "prithivMLmods/Electric-Blue-Flux-LoRA", | |
| "weights": "Electric-Blue.safetensors", | |
| "trigger_word": "Electric Blue" | |
| }, | |
| #140 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Clouds-Illusion-Flux-LoRA/resolve/main/images/CI2.png", | |
| "title": "Clouds Illusion", | |
| "repo": "prithivMLmods/Clouds-Illusion-Flux-LoRA", | |
| "weights": "Clouds-Illusion.safetensors", | |
| "trigger_word": "Clouds Illusion" | |
| }, | |
| #141 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Digital-Yellow-Flux-LoRA/resolve/main/images/DY3.png", | |
| "title": "Digital Yellow", | |
| "repo": "prithivMLmods/Digital-Yellow-Flux-LoRA", | |
| "weights": "Digital-Yellow.safetensors", | |
| "trigger_word": "Digital Yellow" | |
| }, | |
| #142 | |
| { | |
| "image": "https://huggingface.co/cfahlgren1/flux-qwen-capybara/resolve/main/images/example_72ao6twvk.png", | |
| "title": "Flux Qwen Capybara", | |
| "repo": "cfahlgren1/flux-qwen-capybara", | |
| "weights": "flux-qwen-capybara.safetensors", | |
| "trigger_word": "QWENCAPY" | |
| }, | |
| #143 | |
| { | |
| "image": "https://huggingface.co/dasdsff/PleinAirArt/resolve/main/images/e7499ccc-7504-4086-842f-275a5428ef0e.jpg", | |
| "title": "Plein Air Art ", | |
| "repo": "dasdsff/PleinAirArt", | |
| "weights": "PleinAir_000002500.safetensors", | |
| "trigger_word": "P1e!n" | |
| }, | |
| #144 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Orange-Chroma-Flux-LoRA/resolve/main/images/OC1.png", | |
| "title": "Orange Chroma", | |
| "repo": "prithivMLmods/Orange-Chroma-Flux-LoRA", | |
| "weights": "Orange-Chroma.safetensors", | |
| "trigger_word": "Orange Chroma" | |
| }, | |
| #145 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Lime-Green-Flux-LoRA/resolve/main/images/LM1.png", | |
| "title": "Lime Green", | |
| "repo": "prithivMLmods/Lime-Green-Flux-LoRA", | |
| "weights": "Lime-Green.safetensors", | |
| "trigger_word": "Lime Green" | |
| }, | |
| #146 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Fractured-Line-Flare/resolve/main/images/FS1.png", | |
| "title": "Line Flare", | |
| "repo": "prithivMLmods/Fractured-Line-Flare", | |
| "weights": "Fractured-Line-Flare.safetensors", | |
| "trigger_word": "Fractured Line Flare" | |
| }, | |
| #147 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Golden-Dust-Flux-LoRA/resolve/main/images/GD2.png", | |
| "title": "Golden Dust", | |
| "repo": "prithivMLmods/Golden-Dust-Flux-LoRA", | |
| "weights": "Golden-Dust.safetensors", | |
| "trigger_word": "Golden Dust" | |
| }, | |
| #148 | |
| { | |
| "image": "https://huggingface.co/prithivMLmods/Purple-Dreamy-Flux-LoRA/resolve/main/images/PD3.png", | |
| "title": "Purple Dream", | |
| "repo": "prithivMLmods/Purple-Dreamy-Flux-LoRA", | |
| "weights": "Purple-Dreamy.safetensors", | |
| "trigger_word": "Purple Dreamy" | |
| }, | |
| #149 | |
| { | |
| "image": "https://huggingface.co/tryonlabs/FLUX.1-dev-LoRA-Outfit-Generator/resolve/main/images/sample7.jpeg", | |
| "title": "Outfit Generator", | |
| "repo": "tryonlabs/FLUX.1-dev-LoRA-Outfit-Generator", | |
| "weights": "outfit-generator.safetensors", | |
| "trigger_word": "Outfit" | |
| }, | |
| #150 | |
| { | |
| "image": "https://huggingface.co/davisbro/half_illustration/resolve/main/images/example1.webp", | |
| "title": "Half Illustration", | |
| "repo": "davisbro/half_illustration", | |
| "weights": "flux_train_replicate.safetensors", | |
| "trigger_word": "in the style of TOK" | |
| }, | |
| #151 | |
| { | |
| "image": "https://huggingface.co/bingbangboom/flux_oilscape/resolve/main/extras/3.jpg", | |
| "title": "Oilscape", | |
| "repo": "bingbangboom/flux_oilscape", | |
| "weights": "flux_Oilstyle.safetensors", | |
| "trigger_word": "in the style of Oilstyle002" | |
| } | |
| #add new | |
| ] | |
| #--------------------------------------------------Model Initialization-----------------------------------------------------------------------------------------# | |
| dtype = torch.bfloat16 | |
| device = "cuda" if torch.cuda.is_available() else "cpu" | |
| base_model = "black-forest-labs/FLUX.1-dev" | |
| #TAEF1 is very tiny autoencoder which uses the same "latent API" as FLUX.1's VAE. FLUX.1 is useful for real-time previewing of the FLUX.1 generation process.# | |
| taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to(device) | |
| good_vae = AutoencoderKL.from_pretrained(base_model, subfolder="vae", torch_dtype=dtype).to(device) | |
| pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype, vae=taef1).to(device) | |
| pipe_i2i = AutoPipelineForImage2Image.from_pretrained(base_model, | |
| vae=good_vae, | |
| transformer=pipe.transformer, | |
| text_encoder=pipe.text_encoder, | |
| tokenizer=pipe.tokenizer, | |
| text_encoder_2=pipe.text_encoder_2, | |
| tokenizer_2=pipe.tokenizer_2, | |
| torch_dtype=dtype | |
| ) | |
| MAX_SEED = 2**32-1 | |
| pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe) | |
| class calculateDuration: | |
| def __init__(self, activity_name=""): | |
| self.activity_name = activity_name | |
| def __enter__(self): | |
| self.start_time = time.time() | |
| return self | |
| def __exit__(self, exc_type, exc_value, traceback): | |
| self.end_time = time.time() | |
| self.elapsed_time = self.end_time - self.start_time | |
| if self.activity_name: | |
| print(f"Elapsed time for {self.activity_name}: {self.elapsed_time:.6f} seconds") | |
| else: | |
| print(f"Elapsed time: {self.elapsed_time:.6f} seconds") | |
| def update_selection(evt: gr.SelectData, width, height): | |
| selected_lora = loras[evt.index] | |
| new_placeholder = f"Type a prompt for {selected_lora['title']}" | |
| lora_repo = selected_lora["repo"] | |
| updated_text = f"### Selected: [{lora_repo}](https://huggingface.co/{lora_repo}) ✅" | |
| if "aspect" in selected_lora: | |
| if selected_lora["aspect"] == "portrait": | |
| width = 768 | |
| height = 1024 | |
| elif selected_lora["aspect"] == "landscape": | |
| width = 1024 | |
| height = 768 | |
| else: | |
| width = 1024 | |
| height = 1024 | |
| return ( | |
| gr.update(placeholder=new_placeholder), | |
| updated_text, | |
| evt.index, | |
| width, | |
| height, | |
| ) | |
| def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scale, progress): | |
| pipe.to("cuda") | |
| generator = torch.Generator(device="cuda").manual_seed(seed) | |
| with calculateDuration("Generating image"): | |
| # Generate image | |
| for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images( | |
| prompt=prompt_mash, | |
| num_inference_steps=steps, | |
| guidance_scale=cfg_scale, | |
| width=width, | |
| height=height, | |
| generator=generator, | |
| joint_attention_kwargs={"scale": lora_scale}, | |
| output_type="pil", | |
| good_vae=good_vae, | |
| ): | |
| yield img | |
| def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps, cfg_scale, width, height, lora_scale, seed): | |
| generator = torch.Generator(device="cuda").manual_seed(seed) | |
| pipe_i2i.to("cuda") | |
| image_input = load_image(image_input_path) | |
| final_image = pipe_i2i( | |
| prompt=prompt_mash, | |
| image=image_input, | |
| strength=image_strength, | |
| num_inference_steps=steps, | |
| guidance_scale=cfg_scale, | |
| width=width, | |
| height=height, | |
| generator=generator, | |
| joint_attention_kwargs={"scale": lora_scale}, | |
| output_type="pil", | |
| ).images[0] | |
| return final_image | |
| def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)): | |
| if selected_index is None: | |
| raise gr.Error("You must select a LoRA before proceeding.🧨") | |
| selected_lora = loras[selected_index] | |
| lora_path = selected_lora["repo"] | |
| trigger_word = selected_lora["trigger_word"] | |
| if(trigger_word): | |
| if "trigger_position" in selected_lora: | |
| if selected_lora["trigger_position"] == "prepend": | |
| prompt_mash = f"{trigger_word} {prompt}" | |
| else: | |
| prompt_mash = f"{prompt} {trigger_word}" | |
| else: | |
| prompt_mash = f"{trigger_word} {prompt}" | |
| else: | |
| prompt_mash = prompt | |
| with calculateDuration("Unloading LoRA"): | |
| pipe.unload_lora_weights() | |
| pipe_i2i.unload_lora_weights() | |
| #LoRA weights flow | |
| with calculateDuration(f"Loading LoRA weights for {selected_lora['title']}"): | |
| pipe_to_use = pipe_i2i if image_input is not None else pipe | |
| weight_name = selected_lora.get("weights", None) | |
| pipe_to_use.load_lora_weights( | |
| lora_path, | |
| weight_name=weight_name, | |
| low_cpu_mem_usage=True | |
| ) | |
| with calculateDuration("Randomizing seed"): | |
| if randomize_seed: | |
| seed = random.randint(0, MAX_SEED) | |
| if(image_input is not None): | |
| final_image = generate_image_to_image(prompt_mash, image_input, image_strength, steps, cfg_scale, width, height, lora_scale, seed) | |
| yield final_image, seed, gr.update(visible=False) | |
| else: | |
| image_generator = generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scale, progress) | |
| final_image = None | |
| step_counter = 0 | |
| for image in image_generator: | |
| step_counter+=1 | |
| final_image = image | |
| progress_bar = f'<div class="progress-container"><div class="progress-bar" style="--current: {step_counter}; --total: {steps};"></div></div>' | |
| yield image, seed, gr.update(value=progress_bar, visible=True) | |
| yield final_image, seed, gr.update(value=progress_bar, visible=False) | |
| def get_huggingface_safetensors(link): | |
| split_link = link.split("/") | |
| if(len(split_link) == 2): | |
| model_card = ModelCard.load(link) | |
| base_model = model_card.data.get("base_model") | |
| print(base_model) | |
| #Allows Both | |
| if((base_model != "black-forest-labs/FLUX.1-dev") and (base_model != "black-forest-labs/FLUX.1-schnell")): | |
| raise Exception("Flux LoRA Not Found!") | |
| # Only allow "black-forest-labs/FLUX.1-dev" | |
| #if base_model != "black-forest-labs/FLUX.1-dev": | |
| #raise Exception("Only FLUX.1-dev is supported, other LoRA models are not allowed!") | |
| image_path = model_card.data.get("widget", [{}])[0].get("output", {}).get("url", None) | |
| trigger_word = model_card.data.get("instance_prompt", "") | |
| image_url = f"https://huggingface.co/{link}/resolve/main/{image_path}" if image_path else None | |
| fs = HfFileSystem() | |
| try: | |
| list_of_files = fs.ls(link, detail=False) | |
| for file in list_of_files: | |
| if(file.endswith(".safetensors")): | |
| safetensors_name = file.split("/")[-1] | |
| if (not image_url and file.lower().endswith((".jpg", ".jpeg", ".png", ".webp"))): | |
| image_elements = file.split("/") | |
| image_url = f"https://huggingface.co/{link}/resolve/main/{image_elements[-1]}" | |
| except Exception as e: | |
| print(e) | |
| gr.Warning(f"You didn't include a link neither a valid Hugging Face repository with a *.safetensors LoRA") | |
| raise Exception(f"You didn't include a link neither a valid Hugging Face repository with a *.safetensors LoRA") | |
| return split_link[1], link, safetensors_name, trigger_word, image_url | |
| def check_custom_model(link): | |
| if(link.startswith("https://")): | |
| if(link.startswith("https://huggingface.co") or link.startswith("https://www.huggingface.co")): | |
| link_split = link.split("huggingface.co/") | |
| return get_huggingface_safetensors(link_split[1]) | |
| else: | |
| return get_huggingface_safetensors(link) | |
| def add_custom_lora(custom_lora): | |
| global loras | |
| if(custom_lora): | |
| try: | |
| title, repo, path, trigger_word, image = check_custom_model(custom_lora) | |
| print(f"Loaded custom LoRA: {repo}") | |
| card = f''' | |
| <div class="custom_lora_card"> | |
| <span>Loaded custom LoRA:</span> | |
| <div class="card_internal"> | |
| <img src="{image}" /> | |
| <div> | |
| <h3>{title}</h3> | |
| <small>{"Using: <code><b>"+trigger_word+"</code></b> as the trigger word" if trigger_word else "No trigger word found. If there's a trigger word, include it in your prompt"}<br></small> | |
| </div> | |
| </div> | |
| </div> | |
| ''' | |
| existing_item_index = next((index for (index, item) in enumerate(loras) if item['repo'] == repo), None) | |
| if(not existing_item_index): | |
| new_item = { | |
| "image": image, | |
| "title": title, | |
| "repo": repo, | |
| "weights": path, | |
| "trigger_word": trigger_word | |
| } | |
| print(new_item) | |
| existing_item_index = len(loras) | |
| loras.append(new_item) | |
| return gr.update(visible=True, value=card), gr.update(visible=True), gr.Gallery(selected_index=None), f"Custom: {path}", existing_item_index, trigger_word | |
| except Exception as e: | |
| gr.Warning(f"Invalid LoRA: either you entered an invalid link, or a non-FLUX LoRA") | |
| return gr.update(visible=True, value=f"Invalid LoRA: either you entered an invalid link, a non-FLUX LoRA"), gr.update(visible=False), gr.update(), "", None, "" | |
| else: | |
| return gr.update(visible=False), gr.update(visible=False), gr.update(), "", None, "" | |
| def remove_custom_lora(): | |
| return gr.update(visible=False), gr.update(visible=False), gr.update(), "", None, "" | |
| run_lora.zerogpu = True | |
| css = ''' | |
| #gen_btn{height: 100%} | |
| #gen_column{align-self: stretch} | |
| #title{text-align: center} | |
| #title h1{font-size: 3em; display:inline-flex; align-items:center} | |
| #title img{width: 100px; margin-right: 0.5em} | |
| #gallery .grid-wrap{height: 10vh} | |
| #lora_list{background: var(--block-background-fill);padding: 0 1em .3em; font-size: 90%} | |
| .card_internal{display: flex;height: 100px;margin-top: .5em} | |
| .card_internal img{margin-right: 1em} | |
| .styler{--form-gap-width: 0px !important} | |
| #progress{height:30px} | |
| #progress .generating{display:none} | |
| .progress-container {width: 100%;height: 30px;background-color: #f0f0f0;border-radius: 15px;overflow: hidden;margin-bottom: 20px} | |
| .progress-bar {height: 100%;background-color: #4f46e5;width: calc(var(--current) / var(--total) * 100%);transition: width 0.5s ease-in-out} | |
| ''' | |
| with gr.Blocks(theme="prithivMLmods/Minecraft-Theme", css=css, delete_cache=(60, 60)) as app: | |
| title = gr.HTML( | |
| """<h1>FLUX LoRA DLC🥳</h1>""", | |
| elem_id="title", | |
| ) | |
| selected_index = gr.State(None) | |
| with gr.Row(): | |
| with gr.Column(scale=3): | |
| prompt = gr.Textbox(label="Prompt", lines=1, placeholder=":/ choose the LoRA and type the prompt ") | |
| with gr.Column(scale=1, elem_id="gen_column"): | |
| generate_button = gr.Button("Generate", variant="primary", elem_id="gen_btn") | |
| with gr.Row(): | |
| with gr.Column(): | |
| selected_info = gr.Markdown("") | |
| gallery = gr.Gallery( | |
| [(item["image"], item["title"]) for item in loras], | |
| label="LoRA DLC's", | |
| allow_preview=False, | |
| columns=3, | |
| elem_id="gallery", | |
| show_share_button=False | |
| ) | |
| with gr.Group(): | |
| custom_lora = gr.Textbox(label="Enter Custom LoRA", placeholder="prithivMLmods/Canopus-LoRA-Flux-Anime") | |
| gr.Markdown("[Check the list of FLUX LoRA's](https://huggingface.co/models?other=base_model:adapter:black-forest-labs/FLUX.1-dev)", elem_id="lora_list") | |
| custom_lora_info = gr.HTML(visible=False) | |
| custom_lora_button = gr.Button("Remove custom LoRA", visible=False) | |
| with gr.Column(): | |
| progress_bar = gr.Markdown(elem_id="progress",visible=False) | |
| result = gr.Image(label="Generated Image") | |
| with gr.Row(): | |
| with gr.Accordion("Advanced Settings", open=False): | |
| with gr.Row(): | |
| input_image = gr.Image(label="Input image", type="filepath") | |
| image_strength = gr.Slider(label="Denoise Strength", info="Lower means more image influence", minimum=0.1, maximum=1.0, step=0.01, value=0.75) | |
| with gr.Column(): | |
| with gr.Row(): | |
| cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, step=0.5, value=3.5) | |
| steps = gr.Slider(label="Steps", minimum=1, maximum=50, step=1, value=28) | |
| with gr.Row(): | |
| width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024) | |
| height = gr.Slider(label="Height", minimum=256, maximum=1536, step=64, value=1024) | |
| with gr.Row(): | |
| randomize_seed = gr.Checkbox(True, label="Randomize seed") | |
| seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True) | |
| lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=3, step=0.01, value=0.95) | |
| gallery.select( | |
| update_selection, | |
| inputs=[width, height], | |
| outputs=[prompt, selected_info, selected_index, width, height] | |
| ) | |
| custom_lora.input( | |
| add_custom_lora, | |
| inputs=[custom_lora], | |
| outputs=[custom_lora_info, custom_lora_button, gallery, selected_info, selected_index, prompt] | |
| ) | |
| custom_lora_button.click( | |
| remove_custom_lora, | |
| outputs=[custom_lora_info, custom_lora_button, gallery, selected_info, selected_index, custom_lora] | |
| ) | |
| gr.on( | |
| triggers=[generate_button.click, prompt.submit], | |
| fn=run_lora, | |
| inputs=[prompt, input_image, image_strength, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale], | |
| outputs=[result, seed, progress_bar] | |
| ) | |
| app.queue() | |
| app.launch() |