Add python-dotenv

ECMWFCode4Earth · Oct 24, 2023 · f335b0c · f335b0c
1 parent cd94a8f
commit f335b0c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/deepr/model/diffusion_trainer.py b/deepr/model/diffusion_trainer.py
@@ -7,6 +7,7 @@
 import torch
 import torch.nn.functional as F
 from accelerate import Accelerator
+from dotenv import find_dotenv, load_dotenv
 from huggingface_hub import Repository
 from tqdm import tqdm
 from transformers import get_cosine_schedule_with_warmup
@@ -18,6 +19,8 @@
 
 logger = get_logger(__name__)
 
+load_dotenv(find_dotenv())
+
 
 def train_diffusion(
     config: TrainingConfig,