add `HF_AUTH_TOKEN`

climategan_wrapper.py

@@ -1,6 +1,7 @@
 # based on https://huggingface.co/spaces/NimaBoscarino/climategan/blob/main/inferences.py # noqa: E501
 # thank you @NimaBoscarino
 
+import os
 import re
 from pathlib import Path
 from uuid import uuid4
@@ -152,6 +153,7 @@ class ClimateGAN:
                 revision="fp16",
                 torch_dtype=torch.float16,
                 safety_checker=None,
+                use_auth_token=os.environ.get("HF_AUTH_TOKEN"),
             ).to(self.trainer.device)
             self._stable_diffusion_is_setup = True
         except Exception as e: