-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathrequirements.linux.txt
27 lines (19 loc) · 1.45 KB
/
requirements.linux.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
#! Python3.9
numpy==1.23 # Installing NumPy, a package for scientific computing
--extra-index-url https://download.pytorch.org/whl/cpu
torch==2.1.2+cpu # Installing PyTorch, for Tensor computation and Deep neural networks
# https://github.com/huggingface/alignment-handbook/issues/94
# huggingface_hub>=0.19.2
# "tokenizers>=0.11.1,!=0.11.3,<0.14 is required for a normal functioning of this module" https://stackoverflow.com/a/77702552
# Except this fails.
tokenizers>=0.11.1,!=0.11.3,<0.14 # Installing Tokenizers
# transformers # Installing transformers for State-of-the-art machine Learning for JAX, PyTorch and TensorFlow
# Solves: module 'torch.utils._pytree' has no attribute 'register_pytree_node' https://stackoverflow.com/a/78416003
transformers==4.28.0 # Installing transformers for State-of-the-art machine Learning for JAX, PyTorch and TensorFlow
# https://pypi.org/project/optimum/
# optimum[openvino, nncf] # Installing Optimum to allow OpenVINO and the Neural Network Compression Framework integration
# optimum[openvino] # Installing Optimum to allow OpenVINO integration
optimum[intel] # Installing Optimum to make the model happy
diffusers[torch] # Installing Diffusers for state-of-the-art diffusion in PyTorch and JAX.
CodeProject-AI-SDK # Installing the CodeProject.AI SDK
# last line empty.