Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -72,7 +72,7 @@ def load_lora_pipeline(revision: str):
|
|
| 72 |
pipe.load_lora_weights(lora_path)
|
| 73 |
return pipe
|
| 74 |
|
| 75 |
-
def init_pipelines(style_intensity: str = "
|
| 76 |
global pipeline_base, pipeline_lora
|
| 77 |
revision_map = {
|
| 78 |
"low": "step_750",
|
|
@@ -90,7 +90,7 @@ def init_pipelines(style_intensity: str = "medium"):
|
|
| 90 |
logger.info("LoRA pipeline loaded")
|
| 91 |
|
| 92 |
# Initial pipelines
|
| 93 |
-
init_pipelines("
|
| 94 |
|
| 95 |
def update_lora_pipeline(style_intensity):
|
| 96 |
logger.info(f"Updating LoRA pipeline to style: {style_intensity}")
|
|
@@ -140,7 +140,7 @@ with gr.Blocks() as demo:
|
|
| 140 |
style_intensity = gr.Dropdown(
|
| 141 |
label="LoRA Style Intensity",
|
| 142 |
choices=["low", "medium", "high"],
|
| 143 |
-
value="
|
| 144 |
interactive=True
|
| 145 |
)
|
| 146 |
|
|
|
|
| 72 |
pipe.load_lora_weights(lora_path)
|
| 73 |
return pipe
|
| 74 |
|
| 75 |
+
def init_pipelines(style_intensity: str = "low"):
|
| 76 |
global pipeline_base, pipeline_lora
|
| 77 |
revision_map = {
|
| 78 |
"low": "step_750",
|
|
|
|
| 90 |
logger.info("LoRA pipeline loaded")
|
| 91 |
|
| 92 |
# Initial pipelines
|
| 93 |
+
init_pipelines("low")
|
| 94 |
|
| 95 |
def update_lora_pipeline(style_intensity):
|
| 96 |
logger.info(f"Updating LoRA pipeline to style: {style_intensity}")
|
|
|
|
| 140 |
style_intensity = gr.Dropdown(
|
| 141 |
label="LoRA Style Intensity",
|
| 142 |
choices=["low", "medium", "high"],
|
| 143 |
+
value="low",
|
| 144 |
interactive=True
|
| 145 |
)
|
| 146 |
|