Skip to content

Commit

Permalink
Fix for str_to_bool definition in testing utils (huggingface#7461)
Browse files Browse the repository at this point in the history
update
  • Loading branch information
DN6 authored Mar 25, 2024
1 parent 9a34953 commit 3bce0f3
Showing 1 changed file with 16 additions and 16 deletions.
32 changes: 16 additions & 16 deletions src/diffusers/utils/testing_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -141,6 +141,22 @@ def get_tests_dir(append_path=None):
return tests_dir


# Taken from the following PR:
# https://github.com/huggingface/accelerate/pull/1964
def str_to_bool(value) -> int:
"""
Converts a string representation of truth to `True` (1) or `False` (0).
True values are `y`, `yes`, `t`, `true`, `on`, and `1`; False value are `n`, `no`, `f`, `false`, `off`, and `0`;
"""
value = value.lower()
if value in ("y", "yes", "t", "true", "on", "1"):
return 1
elif value in ("n", "no", "f", "false", "off", "0"):
return 0
else:
raise ValueError(f"invalid truth value {value}")


def parse_flag_from_env(key, default=False):
try:
value = os.environ[key]
Expand Down Expand Up @@ -920,22 +936,6 @@ def backend_supports_training(device: str):
return BACKEND_SUPPORTS_TRAINING[device]


# Taken from the following PR:
# https://github.com/huggingface/accelerate/pull/1964
def str_to_bool(value) -> int:
"""
Converts a string representation of truth to `True` (1) or `False` (0).
True values are `y`, `yes`, `t`, `true`, `on`, and `1`; False value are `n`, `no`, `f`, `false`, `off`, and `0`;
"""
value = value.lower()
if value in ("y", "yes", "t", "true", "on", "1"):
return 1
elif value in ("n", "no", "f", "false", "off", "0"):
return 0
else:
raise ValueError(f"invalid truth value {value}")


# Guard for when Torch is not available
if is_torch_available():
# Update device function dict mapping
Expand Down

0 comments on commit 3bce0f3

Please sign in to comment.