Spaces:
Running
Running
Yaron Koresh
commited on
Update app.py
Browse files
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=
|
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 |
|