-
Notifications
You must be signed in to change notification settings - Fork 4
Expand file tree
/
Copy pathrequirements.txt
More file actions
20 lines (14 loc) · 906 Bytes
/
requirements.txt
File metadata and controls
20 lines (14 loc) · 906 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
#! Python3.10
# For Phi-3 ONNX / CPU
numpy # Installing NumPy, a package for scientific computing
--pre
onnxruntime-genai # Installing onnxruntime-genai, the ONNX Runtime generate() API
# torch # Installing PyTorch, an open source machine learning framework
# torchvision # Installing TorchVision, for working with computer vision models
# transformers # Installing transformers for State-of-the-art machine Learning for JAX, PyTorch and TensorFlow
# accelerate==0.30.1 # Installing accelerate, the library for handling multi-GPUs/TPU/fp16 with minimal code
# bitsandbytes==0.43.1 # Installing bitsandbytes, a library of k-bit optimizers and matrix multiplication routines
# flash_attn==2.5.8
# openai==1.30.3
CodeProject-AI-SDK # Installing the CodeProject.AI SDK
# last line empty