Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ from KandiSuperRes import get_SR_pipeline
|
|
7 |
|
8 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
9 |
sr_pipe2x = get_SR_pipeline(device=device, fp16=True, flash=True, scale=2)
|
10 |
-
sr_pipe4x = get_SR_pipeline(device=device, fp16=True, flash=
|
11 |
|
12 |
|
13 |
@spaces.GPU()
|
|
|
7 |
|
8 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
9 |
sr_pipe2x = get_SR_pipeline(device=device, fp16=True, flash=True, scale=2)
|
10 |
+
sr_pipe4x = get_SR_pipeline(device=device, fp16=True, flash=False, scale=4)
|
11 |
|
12 |
|
13 |
@spaces.GPU()
|