Spaces:
mikitona
/
Running on Zero

mikitona commited on
Commit
13c074c
·
verified ·
1 Parent(s): aa0358e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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=False)#False
39
- parser.add_argument("--use_tile_vae", action='store_true', default=True)#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=False)
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: