Update app.py

app.py

@@ -63,6 +63,7 @@ pipe = DiffusionPipeline.from_pretrained(model_repo_id, unet=unet, torch_dtype=t
 pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
 pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taesdxl", torch_dtype=torch_dtype).to(device)
 
+pipe = pipe.to(torch_dtype)
 unet = pipe.unet
 
 ## Change these parameters based on how you trained your sliderspace sliders
@@ -92,7 +93,7 @@ base_model_id = "black-forest-labs/FLUX.1-schnell"
 max_sequence_length = 256
 flux_pipe = FluxPipeline.from_pretrained(base_model_id, torch_dtype=torch_dtype)
 flux_pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch_dtype)
-flux_pipe = flux_pipe.to(device)
+flux_pipe = flux_pipe.to(device).to(torch_dtype)
 # pipe.enable_sequential_cpu_offload()
 transformer = flux_pipe.transformer
 
@@ -239,6 +240,15 @@ with gr.Blocks(css=css) as demo:
         💻 [GitHub Code](https://github.com/rohitgandikota/sliderspace)
         """)
 
+        gr.Markdown("""
+        User Guide: We released sliderspace directions for a few concepts. To test them: <br>
+        1. Choose the concept from the `SliderSpace trained on` menu. (eg. "Wizard")
+        2. Select one of the many directions we discovered for that concept from the `Discovered Directions` menu. (eg. "Direction 3")
+        3. Choose a slider weight between -1 to 1 (recommended). But feel free to explore the scale (eg. +1.5)
+        4. Finally, write a prompt for the concept you chose. (eg. "picture of a wizard fighting a magical war")
+        5. Click Generate and discover what the direction is !! 
+        """)
+
         with gr.Row():
             prompt = gr.Text(
                 label="Prompt",
@@ -260,7 +270,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             slider_space = gr.Dropdown(
                 choices=SDXL_CONCEPTS,
-                label="SliderSpace Concept",
+                label="SliderSpace trained on",
                 value=SDXL_CONCEPTS[0]
             )
             discovered_directions = gr.Dropdown(