|
|
@ -105,7 +105,7 @@ setup(
|
|
|
|
"torchdiffeq>=0.2.0",
|
|
|
|
"torchdiffeq>=0.2.0",
|
|
|
|
"torchvision>=0.13.1",
|
|
|
|
"torchvision>=0.13.1",
|
|
|
|
"transformers>=4.19.2",
|
|
|
|
"transformers>=4.19.2",
|
|
|
|
"triton>=2.0.0; sys_platform!='darwin' and platform_machine!='aarch64'",
|
|
|
|
"triton>=2.0.0; sys_platform!='darwin' and platform_machine!='aarch64' and sys_platform == 'linux'",
|
|
|
|
"kornia>=0.6",
|
|
|
|
"kornia>=0.6",
|
|
|
|
"uvicorn>=0.16.0",
|
|
|
|
"uvicorn>=0.16.0",
|
|
|
|
# "xformers>=0.0.22; sys_platform!='darwin' and platform_machine!='aarch64'",
|
|
|
|
# "xformers>=0.0.22; sys_platform!='darwin' and platform_machine!='aarch64'",
|
|
|
|