Spaces:
Running
on
Zero
Running
on
Zero
rem one diff
Browse files- check_app.py +1 -6
check_app.py
CHANGED
|
@@ -53,12 +53,7 @@ MODEL_CONFIGS = {
|
|
| 53 |
"repo_id": "Alpha-VLLM/Lumina-Next-SFT-diffusers",
|
| 54 |
"pipeline_class": LuminaText2ImgPipeline,
|
| 55 |
"cache_dir": cache_dir,
|
| 56 |
-
}
|
| 57 |
-
"OneDiffusion": {
|
| 58 |
-
"repo_id": "lehduong/OneDiffusion",
|
| 59 |
-
"pipeline_class": OneDiffusionPipeline,
|
| 60 |
-
"cache_dir": cache_dir,
|
| 61 |
-
},
|
| 62 |
}
|
| 63 |
|
| 64 |
def generate_image_with_progress(pipe, prompt, num_steps, guidance_scale=None, seed=None, progress=gr.Progress()):
|
|
|
|
| 53 |
"repo_id": "Alpha-VLLM/Lumina-Next-SFT-diffusers",
|
| 54 |
"pipeline_class": LuminaText2ImgPipeline,
|
| 55 |
"cache_dir": cache_dir,
|
| 56 |
+
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 57 |
}
|
| 58 |
|
| 59 |
def generate_image_with_progress(pipe, prompt, num_steps, guidance_scale=None, seed=None, progress=gr.Progress()):
|