diff --git a/environment.yml b/environment.yml index 364b494..6ad4995 100644 --- a/environment.yml +++ b/environment.yml @@ -7,8 +7,12 @@ channels: - nvidia - conda-forge - pyg + - HuggingFace dependencies: +#---# compiler + - ninja +#---# python libs - python - pytorch - torchvision