Overhaul

Fix #38
Fix #39

LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2021-2023 HolyWu
+Copyright (c) 2021 HolyWu
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

README.md

@@ -3,16 +3,16 @@ Plug-and-Play Image Restoration with Deep Denoiser Prior, based on https://githu
 
 
 ## Dependencies
-- [NumPy](https://numpy.org/install)
-- [PyTorch](https://pytorch.org/get-started) >=2.0.1
-- [VapourSynth](http://www.vapoursynth.com/) >=R60
+- [PyTorch](https://pytorch.org/get-started/) 2.3 or later
+- [VapourSynth](http://www.vapoursynth.com/) R66 or later
 
-`trt` requires additional runtime libraries:
-- [CUDA Toolkit](https://developer.nvidia.com/cuda-toolkit) 11.8
-- [cuDNN](https://developer.nvidia.com/cudnn) 8.9
-- [TensorRT](https://developer.nvidia.com/tensorrt) >=8.6.1.6
+`trt` requires additional Python packages:
+- [TensorRT](https://developer.nvidia.com/tensorrt/) 10.0.1
+- [Torch-TensorRT](https://pytorch.org/TensorRT/)
 
-For ease of installation on Windows, you can download the 7z file on [Releases](https://github.com/HolyWu/vs-dpir/releases) which contains required runtime libraries and Python wheel file. Either add the unzipped directory to your system `PATH` or copy the DLL files to a directory which is already in your system `PATH`. Finally pip install the Python wheel file.
+To install TensorRT, simply run `pip install tensorrt==10.0.1`
+
+To install Torch-TensorRT, Windows users can pip install the whl file on [Releases](https://github.com/HolyWu/vs-dpir/releases). Linux users can run `pip install --pre torch-tensorrt --index-url https://download.pytorch.org/whl/nightly/cu121` (requires PyTorch nightly build).
 
 
 ## Installation

pyproject.toml

@@ -8,26 +8,32 @@ version = "3.1.1"
 description = "DPIR function for VapourSynth"
 readme = "README.md"
 requires-python = ">=3.10"
-license = {file = "LICENSE"}
-authors = [{name = "HolyWu", email = "[email protected]"}]
-keywords = ["DPIR", "VapourSynth"]
+license-files = { paths = ["LICENSE"] }
+authors = [
+  { name = "HolyWu", email = "[email protected]" },
+]
+keywords = [
+  "DPIR",
+  "PyTorch",
+  "TensorRT",
+  "VapourSynth",
+]
 classifiers = [
+  "Environment :: GPU :: NVIDIA CUDA",
   "License :: OSI Approved :: MIT License",
   "Operating System :: OS Independent",
-  "Programming Language :: Python :: 3 :: Only",
-  "Topic :: Multimedia :: Video"
+  "Programming Language :: Python :: 3",
+  "Topic :: Multimedia :: Video",
 ]
 dependencies = [
-  "numpy>=1.24.3",
-  "requests>=2.30.0",
-  "tensorrt>=8.6.1",
-  "torch>=2.0.1",
-  "torch-tensorrt-fx-only>=1.5.0.dev0",
-  "tqdm>=4.65.0",
-  "VapourSynth>=60",
-  "vstools>=2.1.0"
+  "numpy",
+  "requests",
+  "torch>=2.3.0",
+  "tqdm",
+  "VapourSynth>=66",
+  "vstools",
 ]
 
 [project.urls]
-"Homepage" = "https://github.com/HolyWu/vs-dpir"
-"Bug Tracker" = "https://github.com/HolyWu/vs-dpir/issues"
+Homepage = "https://github.com/HolyWu/vs-dpir"
+Issues = "https://github.com/HolyWu/vs-dpir/issues"