Download flux model for now

stablecog · Sep 5, 2024 · b292aeb · b292aeb
1 parent a6825c2
commit b292aeb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/endpoints/flux1/pipe.py b/src/endpoints/flux1/pipe.py
@@ -30,8 +30,8 @@ def get_pipe_object(to_cuda: bool = True) -> Flux1PipeObject:
 
 
 if __name__ == "__main__":
-    """ get_pipe_object(to_cuda=False) """
-    logging.info("🔵 Skip downloading model")
+    get_pipe_object(to_cuda=False)
+    """ logging.info("🔵 Skip downloading model")
     folder = os.environ.get("HF_DATASETS_CACHE", None)
     if folder is not None:
-        os.makedirs(folder, exist_ok=True)
+        os.makedirs(folder, exist_ok=True) """