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
Files changed (1) hide show
  1. app.py +6 -1
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
- "sdxl_lightning_8step_lora_290441",
 
 
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