mirror of
https://github.com/lucidrains/DALLE2-pytorch.git
synced 2025-12-19 01:34:19 +01:00
depend on open-clip-torch (#261)
fix the previous commit which assumes open_clip is installed
This commit is contained in:
1
setup.py
1
setup.py
@@ -26,6 +26,7 @@ setup(
|
|||||||
install_requires=[
|
install_requires=[
|
||||||
'accelerate',
|
'accelerate',
|
||||||
'click',
|
'click',
|
||||||
|
'open-clip-torch>=2.0.0,<3.0.0',
|
||||||
'clip-anytorch>=2.4.0',
|
'clip-anytorch>=2.4.0',
|
||||||
'coca-pytorch>=0.0.5',
|
'coca-pytorch>=0.0.5',
|
||||||
'ema-pytorch>=0.0.7',
|
'ema-pytorch>=0.0.7',
|
||||||
|
|||||||
Reference in New Issue
Block a user