garg-aayush
commited on
Commit
•
352460f
1
Parent(s):
68f8de3
update requirements file, hander with tiling parameter
Browse files- handler.py +1 -1
- requirements.txt +1 -1
handler.py
CHANGED
@@ -25,7 +25,7 @@ class EndpointHandler:
|
|
25 |
model_path=f"/repository/weights/Real-ESRGAN-x4plus.pth",
|
26 |
# dni_weight=dni_weight,
|
27 |
model= RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=4),
|
28 |
-
tile=
|
29 |
tile_pad=10,
|
30 |
# pre_pad=args.pre_pad,
|
31 |
half=True,
|
|
|
25 |
model_path=f"/repository/weights/Real-ESRGAN-x4plus.pth",
|
26 |
# dni_weight=dni_weight,
|
27 |
model= RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=4),
|
28 |
+
tile=1000,
|
29 |
tile_pad=10,
|
30 |
# pre_pad=args.pre_pad,
|
31 |
half=True,
|
requirements.txt
CHANGED
@@ -4,7 +4,7 @@ gfpgan>=1.3.5
|
|
4 |
numpy
|
5 |
opencv-python
|
6 |
Pillow
|
7 |
-
torch
|
8 |
torchvision==0.16.2
|
9 |
tqdm
|
10 |
realesrgan
|
|
|
4 |
numpy
|
5 |
opencv-python
|
6 |
Pillow
|
7 |
+
torch>=1.7
|
8 |
torchvision==0.16.2
|
9 |
tqdm
|
10 |
realesrgan
|