NagisaNao commited on
Commit
28206d7
1 Parent(s): 5b2bf9c

update dep for kaggle

Browse files
files_cells/python/en/downloading_en.py CHANGED
@@ -71,8 +71,8 @@ if not os.path.exists(flag_file):
71
  "xformers": "pip install xformers==0.0.27 --no-deps"
72
  },
73
  "Kaggle": {
74
- "xformers": "pip install xformers==0.0.26.post1",
75
- # "torch": "pip install torch==2.1.2+cu121 torchvision==0.16.2+cu121 torchaudio==2.1.2 --extra-index-url https://download.pytorch.org/whl/cu121",
76
  # "aiohttp": "pip install trash-cli && trash-put /opt/conda/lib/python3.10/site-packages/aiohttp*" # fix install req
77
  }
78
  }
 
71
  "xformers": "pip install xformers==0.0.27 --no-deps"
72
  },
73
  "Kaggle": {
74
+ "xformers": "pip install xformers==0.0.27",
75
+ "torch": "pip install torchvision==0.18.1 torchaudio==2.3.1 open-clip-torch==2.26.1",
76
  # "aiohttp": "pip install trash-cli && trash-put /opt/conda/lib/python3.10/site-packages/aiohttp*" # fix install req
77
  }
78
  }