Skip to content

H
Hydra Node Http
Commit 74a85f59 authored by novelailab's avatar novelailab
Browse files
Options

hm

parent abeea628
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
hydra_node/config.py
View file @ 74a85f59
...@@ -17,6 +17,11 @@ def init_config_model(): ...@@ -17,6 +17,11 @@ def init_config_model():
config.dtype = os.getenv("DTYPE", "float16") config.dtype = os.getenv("DTYPE", "float16")
config.device = os.getenv("DEVICE", "cuda") config.device = os.getenv("DEVICE", "cuda")
config.amp = os.getenv("AMP", False) config.amp = os.getenv("AMP", False)
if config.amp == "1":
config.amp = True
elif config.amp == "0":
config.amp = False
is_dev = "" is_dev = ""
environment = "production" environment = "production"
if os.environ['DEV'] == "True": if os.environ['DEV'] == "True":
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment