k-l-lambda
commited on
Commit
·
b29163f
1
Parent(s):
c8fe0b8
app.py: appended lora model of sdxl_lightning_2step_lora & sdxl_lightning_4step_lora.
Browse files
app.py
CHANGED
@@ -118,7 +118,9 @@ LORA_MODELS = [
|
|
118 |
"ral-wtchz-sdxl_233487",
|
119 |
"sdxl_cute_social_comic-000002_107980",
|
120 |
"sdxl_glass_136034",
|
121 |
-
"
|
|
|
|
|
122 |
"sdxl_offset_example_v10_113006",
|
123 |
"sdxl_wrong_lora",
|
124 |
"xl_more_art-full_v1_113467",
|
@@ -418,6 +420,9 @@ def generate_image (
|
|
418 |
)
|
419 |
|
420 |
print('task_id:', datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), res.task.task_id)
|
|
|
|
|
|
|
421 |
except Exception as e:
|
422 |
raise gr.Error(f'Error: {e}')
|
423 |
|
|
|
118 |
"ral-wtchz-sdxl_233487",
|
119 |
"sdxl_cute_social_comic-000002_107980",
|
120 |
"sdxl_glass_136034",
|
121 |
+
"sdxl_lightning_2step_lora",
|
122 |
+
"sdxl_lightning_4step_lora",
|
123 |
+
"sdxl_lightning_8step_lora",
|
124 |
"sdxl_offset_example_v10_113006",
|
125 |
"sdxl_wrong_lora",
|
126 |
"xl_more_art-full_v1_113467",
|
|
|
420 |
)
|
421 |
|
422 |
print('task_id:', datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), res.task.task_id)
|
423 |
+
|
424 |
+
if not hasattr(res, 'images_encoded'):
|
425 |
+
raise gr.Error(f'Error: {res.task}')
|
426 |
except Exception as e:
|
427 |
raise gr.Error(f'Error: {e}')
|
428 |
|