Update custum_3d_diffusion/trainings/base.py

custum_3d_diffusion/trainings/base.py

@@ -9,13 +9,6 @@ import abc
 from diffusers.utils import make_image_grid
 import numpy as np
 import wandb
-from huggingface_hub import hf_hub_download, model_info
-try:
-    from huggingface_hub import cached_download
-except ImportError:
-    def cached_download(*args, **kwargs):
-        return hf_hub_download(*args, **kwargs)
-
 
 from custum_3d_diffusion.trainings.utils import load_config
 from custum_3d_diffusion.custum_modules.unifield_processor import ConfigurableUNet2DConditionModel, AttnConfig