freedak/timesfm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

changes to __init__

Browse Source
This commit is contained in:
Rajat Sen
2024-09-13 22:31:01 +00:00
parent 9957f483fb
commit 4b626306aa
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/timesfm/__init__.py
+5 -2
View File
@@ -12,11 +12,14 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
"""TimesFM init file.""" """TimesFM init file."""
print(
"TimesFM v1.2.0. See https://github.com/google-research/timesfm/blob/master/README.md for updated APIs."
)
from timesfm.timesfm_base import freq_map, TimesFmCheckpoint, TimesFmHparams, TimesFmBase from timesfm.timesfm_base import freq_map, TimesFmCheckpoint, TimesFmHparams, TimesFmBase
try: try:
print("Loaded Jax TimesFM.")
from timesfm.timesfm_jax import TimesFmJax as TimesFm from timesfm.timesfm_jax import TimesFmJax as TimesFm
from timesfm import data_loader from timesfm import data_loader
except Exception as _: except Exception as _:
print("No pax dependencies installed.") print("Loaded PyTorch TimesFM.")
from timesfm.timesfm_torch import TimesFmTorch as TimesFm from timesfm.timesfm_torch import TimesFmTorch as TimesFm