first commit

app.py

@@ -0,0 +1,255 @@
+import gradio as gr
+import numpy as np
+import random
+from PIL import Image
+import os
+
+# import spaces
+from diffusers import StableDiffusion3Pipeline
+import torch
+from peft import PeftModel
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+model_repo_id = "frankjoshua/stable-diffusion-3.5-medium"
+
+if torch.cuda.is_available():
+    torch_dtype = torch.float16
+else:
+    torch_dtype = torch.float32
+
+pipe = StableDiffusion3Pipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
+
+MAX_SEED = np.iinfo(np.int32).max
+MAX_IMAGE_SIZE = 1024
+
+lora_models = {
+    "None": None,
+    "GenEval": "jieliu/SD3.5M-FlowGRPO-GenEval",
+    "Text Rendering": "jieliu/SD3.5M-FlowGRPO-Text",
+    "Human Prefer": "jieliu/SD3.5M-FlowGRPO-PickScore",
+}
+
+lora_prompts = {
+    "GenEval": os.path.join(os.getcwd(), "prompts/geneval.txt"),
+    "Text Rendering": os.path.join(os.getcwd(), "prompts/ocr.txt"),
+    "Human Prefer": os.path.join(os.getcwd(), "prompts/pickscore.txt"),
+}
+
+pipe.transformer = PeftModel.from_pretrained(pipe.transformer, lora_models["GenEval"], adapter_name="GenEval")
+pipe.transformer.load_adapter(lora_models["Text Rendering"], adapter_name="Text Rendering")
+pipe.transformer.load_adapter(lora_models["Human Prefer"], adapter_name="Human Prefer")
+pipe = pipe.to(device)
+
+COUNTER_FILE = os.path.join(os.getcwd(),"model_call_counter.txt")
+
+def get_call_count():
+    if not os.path.exists(COUNTER_FILE):
+        return 0
+    try:
+        with open(COUNTER_FILE, 'r') as f:
+            return int(f.read().strip())
+    except:
+        return 0
+
+def update_call_count():
+    count = get_call_count() + 1
+    with open(COUNTER_FILE, 'w') as f:
+        f.write(str(count))
+    return count
+
+def sample_prompt(lora_model):
+    if lora_model in lora_models and lora_model != "None":
+        file_path = f"{lora_prompts[lora_model]}"
+        try:
+            with open(file_path, 'r') as file:
+                prompts = file.readlines()
+            return random.choice(prompts).strip()
+        except FileNotFoundError:
+            return "Prompt file not found."
+    return ""
+
+def create_grid_image(images):
+    # Create a 2x2 grid from the 4 images
+    width, height = images[0].size
+    grid_image = Image.new('RGB', (width * 2, height * 2))
+    
+    # Paste images in a 2x2 grid
+    grid_image.paste(images[0], (0, 0))
+    grid_image.paste(images[1], (width, 0))
+    grid_image.paste(images[2], (0, height))
+    grid_image.paste(images[3], (width, height))
+    
+    return grid_image
+
+# @spaces.GPU #[uncomment to use ZeroGPU]
+def infer(
+    prompt,
+    seed,
+    randomize_seed,
+    width,
+    height,
+    guidance_scale,
+    num_inference_steps,
+    lora_model,
+    progress=gr.Progress(track_tqdm=True),
+):
+    call_count = update_call_count()
+    
+    images = []
+    seeds = []
+    
+    # Generate 4 images
+    for i in range(4):
+        if randomize_seed:
+            current_seed = random.randint(0, MAX_SEED)
+        else:
+            current_seed = seed + i  # Use sequential seeds if not randomizing
+        
+        seeds.append(current_seed)
+        generator = torch.Generator().manual_seed(current_seed)
+        sampled_prompt = sample_prompt(lora_model)
+        final_prompt = prompt if prompt else sampled_prompt
+
+        if lora_model == "None":
+            with pipe.transformer.disable_adapter():
+                image = pipe(
+                    prompt=final_prompt,
+                    negative_prompt="",
+                    guidance_scale=guidance_scale,
+                    num_inference_steps=num_inference_steps,
+                    width=width,
+                    height=height,
+                    generator=generator,
+                ).images[0]
+        else:
+            pipe.transformer.set_adapter(lora_model)
+            image = pipe(
+                prompt=final_prompt,
+                negative_prompt="",
+                guidance_scale=guidance_scale,
+                num_inference_steps=num_inference_steps,
+                width=width,
+                height=height,
+                generator=generator,
+            ).images[0]
+        
+        images.append(image)
+    
+    # Create a 2x2 grid from the 4 images
+    grid_image = create_grid_image(images)
+    
+    return grid_image, ", ".join(map(str, seeds)), f"Model has been called {call_count} times"
+
+
+css = """
+#col-container {
+    margin: 0 auto;
+    max-width: 640px;
+}
+"""
+
+with gr.Blocks(css=css) as demo:
+    with gr.Column(elem_id="col-container"):
+        gr.Markdown(" # SD3.5 Medium+Flow-GRPO\n\nOur model is trained separately for different tasks, so it's best to choose the corresponding prompt format for each task.\n\n**User Guide:**\n1. Select a LoRA Model (choose 'None' for the base model)\n2. Click 'Sample Prompt' to randomly select from ~1000 task-specific prompts, or write your own\n3. Click 'Run' to generate images (generates a 2×2 grid of 4 images)")
+
+        with gr.Row():
+            prompt = gr.Textbox(
+                label="Prompt",
+                show_label=False,
+                max_lines=1,
+                placeholder="Enter your prompt",
+                container=False,
+            )
+
+        with gr.Row():
+            lora_model = gr.Dropdown(
+                label="LoRA Model",
+                choices=list(lora_models.keys()),
+                value="None"
+            )
+
+            sample_prompt_button = gr.Button("Sample Prompt", scale=0, variant="secondary")
+
+            def update_sampled_prompt(lora_model):
+                return sample_prompt(lora_model)
+
+            sample_prompt_button.click(
+                fn=update_sampled_prompt,
+                inputs=[lora_model],
+                outputs=[prompt]
+            )
+
+            run_button = gr.Button("Run", scale=0, variant="primary")
+        
+        
+        result = gr.Image(label="Results (2x2 Grid)", show_label=True)
+        seed_display = gr.Textbox(label="Seeds Used", show_label=True)
+        
+        with gr.Accordion("Advanced Settings", open=False):
+            seed = gr.Slider(
+                label="Starting Seed",
+                minimum=0,
+                maximum=MAX_SEED,
+                step=1,
+                value=0,
+            )
+
+            randomize_seed = gr.Checkbox(label="Randomize seeds", value=True)
+
+            with gr.Row():
+                width = gr.Slider(
+                    label="Width",
+                    minimum=256,
+                    maximum=MAX_IMAGE_SIZE,
+                    step=32,
+                    value=512,  # Replace with defaults that work for your model
+                )
+
+                height = gr.Slider(
+                    label="Height",
+                    minimum=256,
+                    maximum=MAX_IMAGE_SIZE,
+                    step=32,
+                    value=512,  # Replace with defaults that work for your model
+                )
+
+            with gr.Row():
+                guidance_scale = gr.Slider(
+                    label="Guidance scale",
+                    minimum=0.0,
+                    maximum=10.0,
+                    step=0.1,
+                    value=4.5,  # Replace with defaults that work for your model
+                )
+
+                num_inference_steps = gr.Slider(
+                    label="Number of inference steps",
+                    minimum=1,
+                    maximum=50,
+                    step=1,
+                    value=40,  # Replace with defaults that work for your model
+                )
+
+        call_count_display = gr.Textbox(
+            label="Model Call Count", 
+            value=f"Model has been called {get_call_count()} times",
+            interactive=False
+        )
+    gr.on(
+        triggers=[run_button.click, prompt.submit],
+        fn=infer,
+        inputs=[
+            prompt,
+            seed,
+            randomize_seed,
+            width,
+            height,
+            guidance_scale,
+            num_inference_steps,
+            lora_model,
+        ],
+        outputs=[result, seed_display, call_count_display],
+    )
+
+if __name__ == "__main__":
+    demo.launch()

model_call_counter.txt

@@ -0,0 +1 @@
+160

prompts/geneval.txt

@@ -0,0 +1,473 @@
+a photo of a bench and a sports ball
+a photo of a toothbrush and a snowboard
+a photo of a toaster and an oven
+a photo of a broccoli and a vase
+a photo of a tennis racket and a wine glass
+a photo of a fork and a knife
+a photo of a hair drier and a cake
+a photo of a horse and a giraffe
+a photo of a horse and a computer keyboard
+a photo of a toothbrush and a carrot
+a photo of a cake and a zebra
+a photo of a hair drier and a bear
+a photo of a knife and a zebra
+a photo of a couch and a wine glass
+a photo of a frisbee and a vase
+a photo of a book and a laptop
+a photo of a dining table and a bear
+a photo of a frisbee and a couch
+a photo of a couch and a horse
+a photo of a toilet and a computer mouse
+a photo of a bottle and a refrigerator
+a photo of a potted plant and a backpack
+a photo of a skateboard and a cake
+a photo of a broccoli and a parking meter
+a photo of a zebra and a bed
+a photo of an oven and a bed
+a photo of a baseball bat and a fork
+a photo of a vase and a spoon
+a photo of a skateboard and a sink
+a photo of a pizza and a bench
+a photo of a bowl and a pizza
+a photo of a tennis racket and a bird
+a photo of a wine glass and a bear
+a photo of a fork and a book
+a photo of a scissors and a bowl
+a photo of a laptop and a carrot
+a photo of a stop sign and a bottle
+a photo of a microwave and a truck
+a photo of a person and a bear
+a photo of a frisbee and a cell phone
+a photo of a parking meter and a teddy bear
+a photo of a tennis racket and a bicycle
+a photo of a stop sign and a motorcycle
+a photo of a fire hydrant and a tennis racket
+a photo of a scissors and a sandwich
+a photo of a pizza and a book
+a photo of a giraffe and a computer mouse
+a photo of a stop sign and a toaster
+a photo of a computer mouse and a zebra
+a photo of a chair and a bench
+a photo of a tv and a carrot
+a photo of a surfboard and a suitcase
+a photo of a computer keyboard and a laptop
+a photo of a computer keyboard and a microwave
+a photo of a scissors and a bird
+a photo of a person and a snowboard
+a photo of a cow and a horse
+a photo of a handbag and a refrigerator
+a photo of a chair and a laptop
+a photo of a toothbrush and a bench
+a photo of a book and a baseball bat
+a photo of a horse and a train
+a photo of a bench and a vase
+a photo of a traffic light and a backpack
+a photo of a sports ball and a cow
+a photo of a computer mouse and a spoon
+a photo of a tv and a bicycle
+a photo of a bench and a snowboard
+a photo of a toothbrush and a toilet
+a photo of a person and an apple
+a photo of a sink and a sports ball
+a photo of a stop sign and a dog
+a photo of a knife and a stop sign
+a photo of a wine glass and a handbag
+a photo of a bowl and a skis
+a photo of a frisbee and an apple
+a photo of a computer keyboard and a cell phone
+a photo of a stop sign and a fork
+a photo of a potted plant and a boat
+a photo of a tv and a cell phone
+a photo of a tie and a broccoli
+a photo of a potted plant and a donut
+a photo of a person and a sink
+a photo of a couch and a snowboard
+a photo of a fork and a baseball glove
+a photo of an apple and a toothbrush
+a photo of a bus and a baseball glove
+a photo of a person and a stop sign
+a photo of a carrot and a couch
+a photo of a baseball bat and a bear
+a photo of a fire hydrant and a train
+a photo of a baseball glove and a carrot
+a photo of a microwave and a bench
+a photo of a cake and a stop sign
+a photo of a car and a computer mouse
+a photo of a suitcase and a dining table
+a photo of a person and a traffic light
+a photo of a cell phone and a horse
+a photo of a baseball bat and a giraffe
+a photo of two clocks
+a photo of two backpacks
+a photo of four handbags
+a photo of two frisbees
+a photo of three sports balls
+a photo of two bears
+a photo of two ties
+a photo of four sinks
+a photo of two toothbrushs
+a photo of three persons
+a photo of three tennis rackets
+a photo of four bowls
+a photo of four vases
+a photo of three cups
+a photo of four computer keyboards
+a photo of three sinks
+a photo of two ovens
+a photo of two toilets
+a photo of two bicycles
+a photo of two trains
+a photo of three oranges
+a photo of three buses
+a photo of three handbags
+a photo of three snowboards
+a photo of two snowboards
+a photo of four dogs
+a photo of three apples
+a photo of two sheeps
+a photo of three hot dogs
+a photo of three zebras
+a photo of three kites
+a photo of four apples
+a photo of three cell phones
+a photo of four baseball gloves
+a photo of three computer keyboards
+a photo of two beds
+a photo of two tv remotes
+a photo of three fire hydrants
+a photo of three books
+a photo of four giraffes
+a photo of two vases
+a photo of four donuts
+a photo of four chairs
+a photo of three baseball bats
+a photo of four stop signs
+a photo of two pizzas
+a photo of three refrigerators
+a photo of two fire hydrants
+a photo of three giraffes
+a photo of four tvs
+a photo of three wine glasses
+a photo of four broccolis
+a photo of three trucks
+a photo of two trucks
+a photo of two carrots
+a photo of two sandwichs
+a photo of four traffic lights
+a photo of four clocks
+a photo of two cars
+a photo of two bananas
+a photo of two wine glasses
+a photo of three pizzas
+a photo of four knifes
+a photo of three suitcases
+a photo of four zebras
+a photo of two teddy bears
+a photo of four skateboards
+a photo of four hot dogs
+a photo of three birds
+a photo of four boats
+a photo of four microwaves
+a photo of two hair driers
+a photo of three laptops
+a photo of three cows
+a photo of two parking meters
+a photo of four benchs
+a photo of three benchs
+a photo of four frisbees
+a photo of four books
+a photo of four buses
+a photo of a blue fire hydrant
+a photo of a pink car
+a photo of a purple cup
+a photo of a blue cow
+a photo of a yellow boat
+a photo of a blue umbrella
+a photo of a blue elephant
+a photo of a yellow elephant
+a photo of a red bicycle
+a photo of a purple suitcase
+a photo of a purple hair drier
+a photo of a white sandwich
+a photo of a purple elephant
+a photo of a green microwave
+a photo of a red zebra
+a photo of a red apple
+a photo of a yellow tv remote
+a photo of a blue toilet
+a photo of an orange orange
+a photo of a black donut
+a photo of a red vase
+a photo of a purple pizza
+a photo of a pink skateboard
+a photo of a green skateboard
+a photo of a purple bear
+a photo of a brown chair
+a photo of a brown computer keyboard
+a photo of an orange cow
+a photo of a brown skis
+a photo of a white kite
+a photo of a red dog
+a photo of a green couch
+a photo of a yellow airplane
+a photo of an orange tv
+a photo of a white scissors
+a photo of a pink cell phone
+a photo of a green surfboard
+a photo of a white fire hydrant
+a photo of a black bicycle
+a photo of a purple carrot
+a photo of a black dining table
+a photo of a purple potted plant
+a photo of a purple backpack
+a photo of a yellow train
+a photo of a pink potted plant
+a photo of a red giraffe
+a photo of a brown bear
+a photo of a black train
+a photo of an orange laptop
+a photo of a green hot dog
+a photo of a yellow parking meter
+a photo of a red potted plant
+a photo of a green traffic light
+a photo of a blue tv
+a photo of a brown refrigerator
+a photo of a black tv remote
+a photo of a purple scissors
+a photo of a yellow orange
+a photo of a brown toaster
+a photo of a red parking meter
+a photo of a brown orange
+a photo of a green clock
+a photo of a white sheep
+a photo of a yellow oven
+a photo of a green vase
+a photo of a black teddy bear
+a photo of a yellow carrot
+a photo of a black hot dog
+a photo of a red scissors
+a photo of a white teddy bear
+a photo of a black skis
+a photo of a blue dining table
+a photo of a black refrigerator
+a photo of a white dog
+a photo of an orange scissors
+a photo of a red cell phone
+a photo of a white orange
+a photo of a blue clock
+a photo of a blue carrot
+a photo of a green motorcycle
+a photo of a pink stop sign
+a photo of a black vase
+a photo of a black backpack
+a photo of a red car
+a photo of a green computer mouse
+a photo of a red backpack
+a photo of a green bus
+a photo of an orange toaster
+a photo of a yellow fork
+a photo of a pink parking meter
+a photo of a blue book
+a photo of a yellow broccoli
+a photo of an orange computer mouse
+a photo of a red cake
+a photo of a dog right of a teddy bear
+a photo of a wine glass above a kite
+a photo of a couch below a cup
+a photo of a laptop left of a cow
+a photo of a fork above a hair drier
+a photo of a tie right of a baseball bat
+a photo of a stop sign above a fork
+a photo of a bird below a skateboard
+a photo of an apple above a tv
+a photo of a train above a potted plant
+a photo of a truck left of a refrigerator
+a photo of a tv remote below a cow
+a photo of a bottle right of a train
+a photo of a dog above a cow
+a photo of a skateboard above a person
+a photo of a baseball glove below an umbrella
+a photo of a dining table right of an oven
+a photo of a hot dog left of a suitcase
+a photo of a bus below a toothbrush
+a photo of a backpack right of a sandwich
+a photo of a cake below a baseball bat
+a photo of a dog right of a tie
+a photo of a suitcase right of a boat
+a photo of a bear above a clock
+a photo of a tv remote left of an umbrella
+a photo of a sports ball left of an umbrella
+a photo of a train right of a dining table
+a photo of a hair drier below an elephant
+a photo of a tennis racket right of a spoon
+a photo of a wine glass right of a hot dog
+a photo of a computer mouse left of a bench
+a photo of a carrot left of an orange
+a photo of a kite above a toothbrush
+a photo of a toaster below a traffic light
+a photo of a cat below a baseball glove
+a photo of a skis right of a zebra
+a photo of a stop sign above a chair
+a photo of a stop sign above a parking meter
+a photo of a hot dog right of a skateboard
+a photo of a pizza below a computer keyboard
+a photo of a hair drier left of a toilet
+a photo of a cow left of a stop sign
+a photo of a suitcase above a skis
+a photo of a book above a laptop
+a photo of a toothbrush below a pizza
+a photo of a toilet left of a kite
+a photo of a tie above a sink
+a photo of a bird left of a couch
+a photo of a bed right of a sports ball
+a photo of an elephant below a surfboard
+a photo of a frisbee right of a motorcycle
+a photo of a vase above a fire hydrant
+a photo of a zebra left of an elephant
+a photo of a bench left of a bear
+a photo of a donut right of a bench
+a photo of a frisbee below a horse
+a photo of a computer keyboard above a snowboard
+a photo of a tv below a cow
+a photo of an elephant below a horse
+a photo of a suitcase left of a banana
+a photo of a train below an airplane
+a photo of a cat below a backpack
+a photo of a backpack below a cake
+a photo of a sandwich below a knife
+a photo of a bicycle above a parking meter
+a photo of a knife right of a suitcase
+a photo of a hot dog above a knife
+a photo of a zebra right of a parking meter
+a photo of a chair left of a zebra
+a photo of a cow below an airplane
+a photo of a cup left of an umbrella
+a photo of a zebra below a computer keyboard
+a photo of a zebra below a broccoli
+a photo of a laptop below a sports ball
+a photo of a truck left of a baseball bat
+a photo of a refrigerator above a baseball bat
+a photo of a tv above a baseball bat
+a photo of a baseball glove right of a bear
+a photo of a refrigerator below a scissors
+a photo of a dining table above a suitcase
+a photo of a parking meter above a broccoli
+a photo of a frisbee above a truck
+a photo of a pizza right of a banana
+a photo of a bus above a boat
+a photo of a cell phone left of a tennis racket
+a photo of a horse right of a broccoli
+a photo of a broccoli above a bottle
+a photo of a vase right of a horse
+a photo of a bear above a spoon
+a photo of a zebra right of a bed
+a photo of a cow right of a laptop
+a photo of a bed right of a frisbee
+a photo of a tie right of a motorcycle
+a photo of a laptop right of a tv
+a photo of a cell phone right of a chair
+a photo of a couch below a potted plant
+a photo of a clock below a tv
+a photo of a couch below a vase
+a photo of a donut below a cat
+a photo of a couch left of a toaster
+a photo of a purple wine glass and a black apple
+a photo of a green bus and a purple microwave
+a photo of a green skis and a brown airplane
+a photo of a yellow computer keyboard and a black sink
+a photo of a pink oven and a green motorcycle
+a photo of a purple parking meter and a red laptop
+a photo of a yellow skateboard and an orange computer mouse
+a photo of a red skis and a brown tie
+a photo of a pink skateboard and a black train
+a photo of a white handbag and a purple bed
+a photo of a purple elephant and a brown sports ball
+a photo of a purple dog and a black dining table
+a photo of a white dining table and a red car
+a photo of a blue cell phone and a green apple
+a photo of a red car and an orange potted plant
+a photo of a brown carrot and a white potted plant
+a photo of a black kite and a green bear
+a photo of a blue laptop and a brown bear
+a photo of a green teddy bear and a brown kite
+a photo of a yellow stop sign and a blue potted plant
+a photo of an orange snowboard and a green cat
+a photo of an orange truck and a pink sink
+a photo of a brown hot dog and a purple pizza
+a photo of a green couch and an orange umbrella
+a photo of a brown bed and a pink cell phone
+a photo of a black broccoli and a yellow cake
+a photo of a red train and a purple bear
+a photo of a purple tennis racket and a black sink
+a photo of a blue vase and a black banana
+a photo of a blue clock and a white cup
+a photo of a red umbrella and a blue couch
+a photo of a white handbag and a red giraffe
+a photo of a pink tv remote and a blue airplane
+a photo of a pink handbag and a black scissors
+a photo of a brown car and a pink hair drier
+a photo of a black bus and a brown cell phone
+a photo of a purple sheep and a pink banana
+a photo of a blue handbag and a white cell phone
+a photo of a white pizza and a green umbrella
+a photo of a white tie and a purple skateboard
+a photo of a yellow sports ball and a green boat
+a photo of a white wine glass and a brown giraffe
+a photo of a yellow bowl and a white baseball glove
+a photo of an orange microwave and a black spoon
+a photo of an orange skateboard and a pink bowl
+a photo of a blue toilet and a white suitcase
+a photo of a white boat and an orange hot dog
+a photo of a yellow dining table and a pink dog
+a photo of a red cake and a purple chair
+a photo of a blue tie and a pink dining table
+a photo of a blue cow and a black computer keyboard
+a photo of a yellow pizza and a green oven
+a photo of a red laptop and a brown car
+a photo of a purple computer keyboard and a blue scissors
+a photo of a green surfboard and an orange oven
+a photo of a yellow parking meter and a pink refrigerator
+a photo of a brown computer mouse and a purple bottle
+a photo of a red umbrella and a green cow
+a photo of a red giraffe and a black cell phone
+a photo of a brown oven and a purple train
+a photo of a blue baseball bat and a pink book
+a photo of a green cup and a yellow bowl
+a photo of a yellow suitcase and a brown bus
+a photo of an orange motorcycle and a pink donut
+a photo of an orange giraffe and a white baseball glove
+a photo of an orange handbag and a green carrot
+a photo of a black bottle and a white refrigerator
+a photo of a white dog and a blue potted plant
+a photo of an orange handbag and a red car
+a photo of a red stop sign and a blue book
+a photo of a yellow car and an orange toothbrush
+a photo of a black potted plant and a yellow toilet
+a photo of a brown dining table and a white suitcase
+a photo of an orange donut and a yellow stop sign
+a photo of a green suitcase and a blue boat
+a photo of an orange tennis racket and a yellow sports ball
+a photo of a purple computer keyboard and a red chair
+a photo of a purple suitcase and an orange pizza
+a photo of a white bottle and a blue sheep
+a photo of a purple backpack and a white umbrella
+a photo of an orange potted plant and a black spoon
+a photo of a green tennis racket and a black dog
+a photo of a yellow handbag and a blue refrigerator
+a photo of a pink broccoli and a red sink
+a photo of a red bowl and a pink sink
+a photo of a white toilet and a red apple
+a photo of a pink dining table and a black sandwich
+a photo of a black car and a green parking meter
+a photo of a yellow bird and a black motorcycle
+a photo of a brown giraffe and a white stop sign
+a photo of a white banana and a black elephant
+a photo of an orange cow and a purple sandwich
+a photo of a red clock and a black cell phone
+a photo of a brown knife and a blue donut
+a photo of a red cup and a pink handbag
+a photo of a yellow bicycle and a red motorcycle
+a photo of a red orange and a purple broccoli
+a photo of an orange traffic light and a white toilet
+a photo of a green cup and a red pizza
+a photo of a blue pizza and a yellow baseball glove

requirements.txt

@@ -0,0 +1,8 @@
+accelerate
+diffusers
+invisible_watermark
+torch
+transformers
+xformers
+peft
+sentencepiece