mirror of https://github.com/vllm-project/vllm
19 lines
550 B
Python
19 lines
550 B
Python
|
import glob
|
||
|
|
|
||
|
|
requires_files = glob.glob('requirements*.txt')
|
||
|
|
requires_files += ["pyproject.toml"]
|
||
|
|
for file in requires_files:
|
||
|
|
print(f">>> cleaning {file}")
|
||
|
|
with open(file, 'r') as f:
|
||
|
|
lines = f.readlines()
|
||
|
|
if "torch" in "".join(lines).lower():
|
||
|
|
print("removed:")
|
||
|
|
with open(file, 'w') as f:
|
||
|
|
for line in lines:
|
||
|
|
if 'torch' not in line.lower():
|
||
|
|
f.write(line)
|
||
|
|
else:
|
||
|
|
print(line.strip())
|
||
|
|
print(f"<<< done cleaning {file}")
|
||
|
|
print()
|