Spaces:
Runtime error
Runtime error
imsuperkong
commited on
Commit
•
384ec30
1
Parent(s):
fe88a60
Update app.py
Browse files
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='
|
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)
|