Pinning PTL to 1.4.2
Signed-off-by: Shrinath Suresh <shrinath@ideas2it.com>
This commit is contained in:
parent
3153d8bef3
commit
8e207f0d3b
|
|
@ -24,7 +24,7 @@ from setuptools import setup, find_packages
|
|||
|
||||
def make_required_install_packages():
|
||||
return [
|
||||
"pytorch-lightning>=1.4.0",
|
||||
"pytorch-lightning==1.4.2",
|
||||
"torch>=1.7.1",
|
||||
"torch-model-archiver",
|
||||
]
|
||||
|
|
|
|||
Loading…
Reference in New Issue