mirror of
https://github.com/huggingface/accelerate.git
synced 2025-10-20 18:13:46 +08:00
a452327e8e
* Working version rebased from main * kwargs * Clean * Fix more nits * Fin * Delay autocast flag * Enable FP8 autocast during eval only if specified * Fin * Rm comment * All done * Zero3 works! * Let the wrapper come off during unwrap_model * Add import check * Migrate all to benchmarks folder and make TE import check work * Add readme * Add README to benchmarks folder * Update CLI to now include fp8 args * Add test config for 0_34 * Finish adding to config yaml * Write docs * Expound docs w/ FP8 * Add to toctree
13 lines
235 B
Docker
13 lines
235 B
Docker
FROM nvcr.io/nvidia/pytorch:24.07-py3
|
|
|
|
RUN pip install transformers evaluate datasets
|
|
RUN git clone https://github.com/huggingface/accelerate.git
|
|
|
|
RUN cd accelerate && \
|
|
pip install -e . && \
|
|
cd benchmarks/fp8
|
|
|
|
RUN /bin/bash
|
|
|
|
|