Yaron Koresh commited on
Commit
4487e3e
·
verified ·
1 Parent(s): 847d40e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -430,7 +430,7 @@ CHECKPOINTS = ESRGANUpscalerCheckpoints(
430
  device = DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
431
  DTYPE = dtype = torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float32
432
 
433
- enhancer = ESRGANUpscaler(checkpoints=CHECKPOINTS, device=device, dtype=DTYPE).to(device=device,dtype=dtype)
434
 
435
  # logging
436
 
 
430
  device = DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
431
  DTYPE = dtype = torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float32
432
 
433
+ enhancer = ESRGANUpscaler(checkpoints=CHECKPOINTS, device=device, dtype=DTYPE)
434
 
435
  # logging
436