mirror of
https://github.com/huggingface/accelerate.git
synced 2025-11-11 22:44:28 +08:00
Compare commits
3 Commits
v0.14-rele
...
v0.6.2
| Author | SHA1 | Date | |
|---|---|---|---|
| a5b8811bfa | |||
| 54a685a92b | |||
| 8bc6c83175 |
2
setup.py
2
setup.py
@ -30,7 +30,7 @@ extras["sagemaker"] = [
|
||||
|
||||
setup(
|
||||
name="accelerate",
|
||||
version="0.7.0.dev0",
|
||||
version="0.6.2",
|
||||
description="Accelerate",
|
||||
long_description=open("README.md", "r", encoding="utf-8").read(),
|
||||
long_description_content_type="text/markdown",
|
||||
|
||||
@ -2,7 +2,7 @@
|
||||
# There's no way to ignore "F401 '...' imported but unused" warnings in this
|
||||
# module, but to preserve other warnings. So, don't check this module at all.
|
||||
|
||||
__version__ = "0.7.0.dev0"
|
||||
__version__ = "0.6.2"
|
||||
|
||||
from .accelerator import Accelerator
|
||||
from .kwargs_handlers import DistributedDataParallelKwargs, GradScalerKwargs, InitProcessGroupKwargs
|
||||
|
||||
@ -56,7 +56,6 @@ def launch_command_parser(subparsers=None):
|
||||
)
|
||||
parser.add_argument(
|
||||
"--mixed_precision",
|
||||
default="no",
|
||||
type=str,
|
||||
choices=["no", "fp16", "bf16"],
|
||||
help="Whether or not to use mixed precision training. "
|
||||
|
||||
Reference in New Issue
Block a user