Fix Huggingface environment variable s|HF_HUB_CACHE_DIR|HF_HUB_CACHE (#1069)

Co-authored-by: Steve McClain <steve@marmalade.co>

Author: charlesfrye

06_gpu_and_ml/sam/segment_anything.py

@@ -62,7 +62,7 @@
 
 
 @app.cls(
-    image=image.env({"HF_HUB_CACHE_DIR": cache_dir}),
+    image=image.env({"HF_HUB_CACHE": cache_dir}),
     volumes={"/root/videos": video_vol, cache_dir: cache_vol},
     gpu="A100",
 )

06_gpu_and_ml/stable_diffusion/flux.py

@@ -57,7 +57,7 @@
         f"git+https://github.com/huggingface/diffusers.git@{diffusers_commit_sha}",
         "numpy<2",
     )
-    .env({"HF_HUB_ENABLE_HF_TRANSFER": "1", "HF_HUB_CACHE_DIR": "/cache"})
+    .env({"HF_HUB_ENABLE_HF_TRANSFER": "1", "HF_HUB_CACHE": "/cache"})
 )
 
 # Later, we'll also use `torch.compile` to increase the speed further.

06_gpu_and_ml/stable_diffusion/image_to_image.py

@@ -40,7 +40,7 @@
     .env(
         {
             "HF_HUB_ENABLE_HF_TRANSFER": "1",  # Allows faster model downloads
-            "HF_HUB_CACHE_DIR": CACHE_DIR,  # Points the Hugging Face cache to a Volume
+            "HF_HUB_CACHE": CACHE_DIR,  # Points the Hugging Face cache to a Volume
         }
     )
 )

06_gpu_and_ml/stable_diffusion/text_to_image.py

@@ -58,7 +58,7 @@
     .env(
         {
             "HF_HUB_ENABLE_HF_TRANSFER": "1",  # faster downloads
-            "HF_HUB_CACHE_DIR": CACHE_DIR,
+            "HF_HUB_CACHE": CACHE_DIR,
         }
     )
 )

misc/trellis3d.py

@@ -110,7 +110,7 @@ def clone_repository():
 
 
 @app.cls(
-    image=trellis_image.env({"HF_HUB_CACHE_DIR": cache_dir}),
+    image=trellis_image.env({"HF_HUB_CACHE": cache_dir}),
     gpu="L4",
     timeout=1 * HOURS,
     container_idle_timeout=1 * MINUTES,