winglian commited on
Commit
7f2027d
1 Parent(s): 8d288a2

tweak for xformers install w pytorch 2.1.0 (#727)

Browse files
Files changed (1) hide show
  1. setup.py +1 -1
setup.py CHANGED
@@ -26,7 +26,7 @@ def parse_requirements():
26
  if "torch==2.1.0" in _install_requires:
27
  _install_requires.pop(_install_requires.index("xformers>=0.0.22"))
28
  _install_requires.append(
29
- "git+https://github.com/facebookresearch/xformers.git@main#egg=xformers"
30
  )
31
 
32
  return _install_requires, _dependency_links
 
26
  if "torch==2.1.0" in _install_requires:
27
  _install_requires.pop(_install_requires.index("xformers>=0.0.22"))
28
  _install_requires.append(
29
+ "xformers @ git+https://github.com/facebookresearch/xformers.git@main"
30
  )
31
 
32
  return _install_requires, _dependency_links