Spaces:
Running
on
Zero
Running
on
Zero
Update cpn.py
Browse files
cpn.py
CHANGED
@@ -36,8 +36,8 @@ class CpnInterface:
|
|
36 |
self.device = ('cuda' if torch.cuda.is_available() else 'cpu') if device is None else device
|
37 |
self.model = cd.models.LitCpn(model).to(device)
|
38 |
self.model.eval()
|
39 |
-
self.tile_size =
|
40 |
-
self.overlap =
|
41 |
|
42 |
def __call__(
|
43 |
self,
|
|
|
36 |
self.device = ('cuda' if torch.cuda.is_available() else 'cpu') if device is None else device
|
37 |
self.model = cd.models.LitCpn(model).to(device)
|
38 |
self.model.eval()
|
39 |
+
self.tile_size = 1536
|
40 |
+
self.overlap = 1152
|
41 |
|
42 |
def __call__(
|
43 |
self,
|