Direct model loading from object storage with Runai Model Streamer (#17948)
Signed-off-by: Noa Neria <noa@run.ai>
This commit is contained in:
@@ -97,6 +97,7 @@ torch = [
|
||||
|
||||
[project.optional-dependencies]
|
||||
checkpoint-engine = ["checkpoint-engine==0.1.2"]
|
||||
runai = ["runai-model-streamer[s3,gcs,azure]>=0.15.7"]
|
||||
diffusion = [
|
||||
"PyYAML==6.0.1",
|
||||
"cloudpickle==3.1.2",
|
||||
@@ -108,7 +109,7 @@ diffusion = [
|
||||
"remote-pdb==2.1.0",
|
||||
"st_attn==0.0.7 ; platform_machine != 'aarch64' and platform_machine != 'arm64'",
|
||||
"vsa==0.0.4 ; platform_machine != 'aarch64' and platform_machine != 'arm64'",
|
||||
"runai_model_streamer>=0.15.5",
|
||||
"runai_model_streamer>=0.15.7",
|
||||
"cache-dit==1.3.0",
|
||||
"addict==2.4.0",
|
||||
"av==16.1.0",
|
||||
|
||||
Reference in New Issue
Block a user