imsuperkong commited on
Commit
384ec30
1 Parent(s): fe88a60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -445,7 +445,7 @@ if __name__ == '__main__':
445
  import argparse
446
  parser = argparse.ArgumentParser()
447
  parser.add_argument('--device', default='cuda')
448
- parser.add_argument('--model_id', default='ckpt/stable-diffusion-2-1-base')
449
  parser.add_argument('--depth_model', default='dpt_beit_large_512', choices=['dpt_beit_large_512', 'dpt_swin2_large_384'])
450
  parser.add_argument('--share', action='store_true')
451
  parser.add_argument('-p', '--port', type=int, default=None)
 
445
  import argparse
446
  parser = argparse.ArgumentParser()
447
  parser.add_argument('--device', default='cuda')
448
+ parser.add_argument('--model_id', default='stabilityai/stable-diffusion-2-1-base')
449
  parser.add_argument('--depth_model', default='dpt_beit_large_512', choices=['dpt_beit_large_512', 'dpt_swin2_large_384'])
450
  parser.add_argument('--share', action='store_true')
451
  parser.add_argument('-p', '--port', type=int, default=None)