Wheelodex » Projects » fastckpt

fastckpt

View on PyPI — Reverse Dependencies (0)

0.0.4 fastckpt-0.0.4-py3-none-any.whl

Wheel Details

Project: fastckpt
Version: 0.0.4
Filename: fastckpt-0.0.4-py3-none-any.whl
Download: [link]
Size: 13902
MD5: bd8b0e70ad4ddb94e946e0b0758bac9a
SHA256: d06ed4b6d967c99ee147fa197032bb0de4ec58e3daf46557ffdda41a1a6d28e7
Uploaded: 2023-10-09 01:00:03 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: fastckpt
Version: 0.0.4
Summary: A fast gradient checkpointing strategy for training with memory-efficient attention (e.g., FlashAttention).
Project-Url: Homepage, https://github.com/RulinShao/fast-ckpt
Project-Url: Bug Tracker, https://github.com/RulinShao/fast-ckpt/issues
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: Apache Software License
Requires-Python: >=3.8
Requires-Dist: aiohttp
Requires-Dist: fastapi
Requires-Dist: httpx
Requires-Dist: markdown2[all]
Requires-Dist: nh3
Requires-Dist: numpy
Requires-Dist: prompt-toolkit (>=3.0.0)
Requires-Dist: pydantic (<2,>=1)
Requires-Dist: requests
Requires-Dist: rich (>=10.0.0)
Requires-Dist: shortuuid
Requires-Dist: tiktoken
Requires-Dist: uvicorn
Requires-Dist: black (==23.3.0); extra == "dev"
Requires-Dist: pylint (==2.8.2); extra == "dev"
Requires-Dist: openai; extra == "llm_judge"
Requires-Dist: anthropic (>=0.3); extra == "llm_judge"
Requires-Dist: ray; extra == "llm_judge"
Requires-Dist: accelerate (>=0.21); extra == "model_worker"
Requires-Dist: peft; extra == "model_worker"
Requires-Dist: sentencepiece; extra == "model_worker"
Requires-Dist: torch; extra == "model_worker"
Requires-Dist: transformers (>=4.31.0); extra == "model_worker"
Requires-Dist: protobuf; extra == "model_worker"
Requires-Dist: einops; extra == "train"
Requires-Dist: flash-attn (>=2.0); extra == "train"
Requires-Dist: wandb; extra == "train"
Requires-Dist: gradio; extra == "webui"
Provides-Extra: dev
Provides-Extra: llm_judge
Provides-Extra: model_worker
Provides-Extra: train
Provides-Extra: webui
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 1341 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.41.2)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
fastckpt/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
fastckpt/llama_flash_attn_ckpt_monkey_patch.py sha256=HXi9e49g14LtlLCwmV4OxDsk5B2ebwDEhjeyOchFbVA 28496
fastckpt/llama_flash_attn_monkey_patch.py sha256=nhFnj-e_asHVIQhb3Toj4eWT6ZAMAIjZ2faGUAY9U_E 4590
fastckpt-0.0.4.dist-info/LICENSE sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE 11356
fastckpt-0.0.4.dist-info/METADATA sha256=DGXG00FFV4qU_PldnwFXGJS5lmqUp_b3-qu89cAdLpk 3019
fastckpt-0.0.4.dist-info/WHEEL sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A 92
fastckpt-0.0.4.dist-info/top_level.txt sha256=Eo6Dehaiqw_CvOOI6rde4-B_cY2JllAev1iYoMkcB6s 9
fastckpt-0.0.4.dist-info/RECORD — —

top_level.txt

fastckpt