AItool commited on
Commit
2b95ee5
·
verified ·
1 Parent(s): 7454a55

Update app_enhance.py

Browse files
Files changed (1) hide show
  1. app_enhance.py +1 -1
app_enhance.py CHANGED
@@ -36,7 +36,7 @@ if not os.path.exists('realesr-general-x4v3.pth'):
36
 
37
  model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=32, upscale=4, act_type='prelu')
38
  model_path = 'realesr-general-x4v3.pth'
39
- half = True if torch.cuda.is_available() else False
40
  upsampler = RealESRGANer(scale=4, model_path=model_path, model=model, tile=0, tile_pad=10, pre_pad=0, half=half)
41
 
42
 
 
36
 
37
  model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=32, upscale=4, act_type='prelu')
38
  model_path = 'realesr-general-x4v3.pth'
39
+ half = True if torch.cuda.is_available() else "cpu"
40
  upsampler = RealESRGANer(scale=4, model_path=model_path, model=model, tile=0, tile_pad=10, pre_pad=0, half=half)
41
 
42