diff --git a/setup.py b/setup.py index 36f30f6e..f825e6db 100644 --- a/setup.py +++ b/setup.py @@ -17,60 +17,73 @@ setup( 'optimizers', "Prompt Engineering" ], - install_requires=[ - 'llama-index', - 'env', - "transformers", - "openai", - "langchain", - "torch==1.13.1", - "torchvision==0.14.1", - "asyncio", - "nest_asyncio", - "bs4", - "playwright", - "duckduckgo_search", - "faiss-cpu", - "python-ptrace==0.9.8", - "wget==3.2", - "accelerate", - "addict", - "albumentations", - "basicsr", - "controlnet-aux", - "diffusers", - "einops", - "gradio", - "imageio", - "imageio-ffmpeg", - "invisible-watermark", - "kornia", - "numpy", - "omegaconf", - "open_clip_torch", - "opencv-python", - "prettytable", - "safetensors", - "streamlit", - "test-tube", - "timm", - "torchmetrics", - "webdataset", - "yapf", - # 'GroundingDINO' - "wolframalpha", - "wikipedia", - "httpx", - "ggl", - "gradio_tools", - "arxiv", - "google-api-python-client", - "google-auth-oauthlib", - "google-auth-httplib2", - "beautifulsoup4", - "O365", - "pytube", - "pydub" + install_requires=[ + 'transformers', + 'openai', + 'langchain', + 'torch==2.0.0', + 'torchvision', + 'asyncio', + 'nest_asyncio', + 'bs4', + 'playwright', + 'duckduckgo_search', + 'faiss-cpu', + 'wget==3.2', + 'accelerate==0.17.1', + 'addict', + 'albumentations', + 'basicsr', + 'controlnet-aux', + 'diffusers==0.14.0', + 'einops', + 'gradio', + 'imageio', + 'imageio-ffmpeg', + 'kornia', + 'numpy', + 'omegaconf', + 'open_clip_torch', + 'opencv-python', + 'prettytable', + 'safetensors', + 'streamlit', + 'test-tube', + 'timm', + 'torchmetrics', + 'webdataset', + 'yapf', + 'wolframalpha', + 'wikipedia==1.4.0', + 'httpx', + 'ggl', + 'gradio_tools', + 'arxiv', + 'google-api-python-client', + 'google-auth-oauth', + 'google-auth-httplib2', + 'beautifulsoup4==4.11.2', + 'O365', + 'pytube', + 'pydub', + 'llama-index', + 'fastapi==0.94.1', + 'pydantic==1.10.6', + 'tenacity==8.2.2', + 'python-dotenv==1.0.0', + 'pillow==9.4.0', + 'boto3==1.26.94', + 'uvicorn==0.21.1', + 'python-ptrace==0.9.8', + 'jinja2==3.1.2', + 'python-multipart==0.0.6', + 'celery==5.2.7', + 'redis==4.5.4', + 'sentencepiece==0.1.97', + 'bitsandbytes==0.37.2', + 'psycopg2-binary==2.9.5', + 'google-search-results==2.4.2', + 'black==23.1.0' ], classifiers=[ 'Development Status :: 4 - Beta',