Add s2s-ft to generate_conda_file.py
This commit is contained in:
Родитель
8072c4074b
Коммит
e9a4a0913b
|
@ -32,22 +32,6 @@
|
|||
"QUICK_RUN = True"
|
||||
]
|
||||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"Manually install the s2s-ft package from the [UniLM repo](https://github.com/microsoft/unilm) by running the following commands in your bash shell. \n",
|
||||
"(We are working on making the s2s-ft package available on PyPI and this step will not be necessary in the future. )\n",
|
||||
"\n",
|
||||
"```\n",
|
||||
"conda activate nlp_gpu\n",
|
||||
"git clone https://github.com/microsoft/unilm.git\n",
|
||||
"git checkout 7f931fcfb965bf60f0fbe2ddd37798f356e6ee5e\n",
|
||||
"cd unilm/s2s-ft\n",
|
||||
"pip install -e .\n",
|
||||
"```"
|
||||
]
|
||||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"metadata": {},
|
||||
|
|
|
@ -11,10 +11,6 @@ ABS_TOL = 0.02
|
|||
|
||||
@pytest.mark.gpu
|
||||
@pytest.mark.integration
|
||||
@pytest.mark.skip(
|
||||
reason="s2s-ft not available on PYPI and can not be included in "
|
||||
"generate_conda_file.py"
|
||||
)
|
||||
def test_unilm_abstractive_summarization(notebooks, tmp):
|
||||
notebook_path = notebooks["unilm_abstractive_summarization"]
|
||||
pm.execute_notebook(
|
||||
|
|
|
@ -96,6 +96,8 @@ PIP_BASE = {
|
|||
"Cython": "Cython>=0.29.13",
|
||||
"googledrivedownloader": "googledrivedownloader>=0.4",
|
||||
"methodtools": "methodtools",
|
||||
"s2s-ft": "-e git+https://github.com/microsoft/unilm.git"
|
||||
"@7f931fcfb965bf60f0fbe2ddd37798f356e6ee5e#egg=s2s-ft&subdirectory=s2s-ft",
|
||||
}
|
||||
|
||||
PIP_GPU = {}
|
||||
|
|
Загрузка…
Ссылка в новой задаче