Update environments and dependencies.

PiperOrigin-RevId: 632263761
This commit is contained in:
Yichen Zhou
2024-05-09 14:05:01 -07:00
committed by siriuz42
parent a4fd4ca7c3
commit b389495851
6 changed files with 62 additions and 20 deletions
+12 -4
View File
@@ -30,9 +30,9 @@ Please look into the README files in the respective benchmark directories within
## Installation
We have two environment files. For GPU installation (assuming CUDA 12 has been
setup), you can create a conda environment `tfm_env` from the base folder
through:
For calling TimesFM, We have two environment files. Inside `timesfm`, for
GPU installation (assuming CUDA 12 has been setup), you can create a conda
environment `tfm_env` from the base folder through:
```
conda env create --file=environment.yml
@@ -53,6 +53,14 @@ pip install -e .
```
to install the package.
**Note**:
1. Running the provided benchmarks would require additional dependencies.
Please use the environment files under `experiments` instead.
2. The dependency `lingvo` does not support `pip install` for macOS. For now you
can manually install it from the source following [these instructions](https://github.com/tensorflow/lingvo).
## Usage
### Initialize the model and load a checkpoint.
@@ -112,7 +120,7 @@ Array inputs, with the frequencies set to low, medium and high respectively.
```python
import numpy as np
forecast_input = [
np.sin(np.linspace(0, 20, 100))
np.sin(np.linspace(0, 20, 100)),
np.sin(np.linspace(0, 20, 200)),
np.sin(np.linspace(0, 20, 400)),
]