English
Inference Endpoints
garg-aayush commited on
Commit
352460f
1 Parent(s): 68f8de3

update requirements file, hander with tiling parameter

Browse files
Files changed (2) hide show
  1. handler.py +1 -1
  2. 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=0,
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==1.7.1
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