Update app.py
Browse files
app.py
CHANGED
@@ -35,16 +35,16 @@ parser.add_argument("--port", type=int, default='6688')
|
|
35 |
parser.add_argument("--no_llava", action='store_true', default=True)#False
|
36 |
parser.add_argument("--use_image_slider", action='store_true', default=False)#False
|
37 |
parser.add_argument("--log_history", action='store_true', default=False)
|
38 |
-
parser.add_argument("--loading_half_params", action='store_true', default=
|
39 |
-
parser.add_argument("--use_tile_vae", action='store_true', default=
|
40 |
parser.add_argument("--encoder_tile_size", type=int, default=512)
|
41 |
parser.add_argument("--decoder_tile_size", type=int, default=64)
|
42 |
-
parser.add_argument("--load_8bit_llava", action='store_true', default=
|
43 |
args = parser.parse_args()
|
44 |
|
45 |
if torch.cuda.device_count() > 0:
|
46 |
SUPIR_device = 'cuda:0'
|
47 |
-
|
48 |
# Load SUPIR
|
49 |
model, default_setting = create_SUPIR_model(args.opt, SUPIR_sign='Q', load_default_setting=True)
|
50 |
if args.loading_half_params:
|
|
|
35 |
parser.add_argument("--no_llava", action='store_true', default=True)#False
|
36 |
parser.add_argument("--use_image_slider", action='store_true', default=False)#False
|
37 |
parser.add_argument("--log_history", action='store_true', default=False)
|
38 |
+
parser.add_argument("--loading_half_params", action='store_true', default=True)#False
|
39 |
+
parser.add_argument("--use_tile_vae", action='store_true', default=False)#False
|
40 |
parser.add_argument("--encoder_tile_size", type=int, default=512)
|
41 |
parser.add_argument("--decoder_tile_size", type=int, default=64)
|
42 |
+
parser.add_argument("--load_8bit_llava", action='store_true', default=True)
|
43 |
args = parser.parse_args()
|
44 |
|
45 |
if torch.cuda.device_count() > 0:
|
46 |
SUPIR_device = 'cuda:0'
|
47 |
+
|
48 |
# Load SUPIR
|
49 |
model, default_setting = create_SUPIR_model(args.opt, SUPIR_sign='Q', load_default_setting=True)
|
50 |
if args.loading_half_params:
|