Yaron Koresh commited on
Commit
9c4ac8a
·
verified ·
1 Parent(s): f130ec5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -431,7 +431,7 @@ CHECKPOINTS = ESRGANUpscalerCheckpoints(
431
  device = DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
432
  DTYPE = dtype = torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float32
433
 
434
- enhancer = ESRGANUpscaler(checkpoints=CHECKPOINTS, device=device, dtype=DTYPE).to(device)
435
 
436
  # logging
437
 
@@ -452,7 +452,7 @@ MAX_SEED = np.iinfo(np.int32).max
452
 
453
  seq=512
454
  image_steps=30
455
- img_accu=9.0
456
 
457
  # ui data
458
 
 
431
  device = DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
432
  DTYPE = dtype = torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float32
433
 
434
+ enhancer = ESRGANUpscaler(checkpoints=CHECKPOINTS, device=device, dtype=DTYPE).to(device=device,dtype=dtype)
435
 
436
  # logging
437
 
 
452
 
453
  seq=512
454
  image_steps=30
455
+ img_accu=10.0
456
 
457
  # ui data
458