Update app.py

app.py

@@ -149,42 +149,60 @@ DownLoad(r"https://civitai.com/api/download/models/39885",str(user_home / r"stab
 DownLoad(r"https://civitai.com/api/download/models/21065",str(user_home / r"stable-diffusion-webui" / r"extensions" / r"sd-webui-additional-networks" / r"models"/ r"lora"),r"LAS.safetensors")
 DownLoad(r"https://civitai.com/api/download/models/39164",str(user_home / r"stable-diffusion-webui" / r"extensions" / r"sd-webui-additional-networks" / r"models"/ r"lora"),r"backlighting.safetensors")
 
+# CPUモード設定（Linux向け修正）
+os.environ["COMMANDLINE_ARGS"] = "--skip-torch-cuda-test --no-half --use-cpu all"
 
 
+# Hugging Face のリポジトリ情報
+HF_REPO = "https://huggingface.co/black-forest-labs/FLUX.1-dev"
+MODEL_NAME = "flux1-dev.safetensors"
+MODEL_DIR = "/home/user/stable-diffusion-webui/models/Stable-diffusion"
+MODEL_PATH = os.path.join(MODEL_DIR, MODEL_NAME)
 
-# Hugging Face トークン取得
+# 環境変数から Hugging Face のトークンを取得
 hf_token = os.getenv("token")
 if not hf_token:
     print("Error: Hugging Face token not found.")
     exit(1)
 
-# 削除処理
-model_path = "/home/user/stable-diffusion-webui/models/Stable-diffusion/flux1-dev.safetensors"
-if os.path.exists(model_path):
-    os.remove(model_path)
+# `git-lfs` がインストールされているか確認
+try:
+    subprocess.run(["git", "lfs", "version"], check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+except subprocess.CalledProcessError:
+    print("Error: Git LFS is not installed. Please install it using:")
+    print("      sudo apt install git-lfs  (Ubuntu/Debian)")
+    print("      brew install git-lfs      (Mac)")
+    print("      choco install git-lfs      (Windows)")
+    exit(1)
 
-# CPUモード設定（Linux向け修正）
-os.environ["COMMANDLINE_ARGS"] = "--skip-torch-cuda-test --no-half --use-cpu all"
+# `git-lfs` を使ってモデルをダウンロード
+if not os.path.exists(MODEL_DIR):
+    os.makedirs(MODEL_DIR, exist_ok=True)
+
+print("Cloning repository using Git LFS...")
+clone_cmd = [
+    "GIT_LFS_SKIP_SMUDGE=1",  # 最初のクローン時にダウンロードをスキップ
+    "git", "clone", HF_REPO, "--depth=1", "--filter=blob:none", MODEL_DIR
+]
+subprocess.run(" ".join(clone_cmd), shell=True, check=True)
 
-# wget でファイルをダウンロード
-cmd = f'wget --header="Authorization: Bearer {hf_token}" -O "{model_path}" "https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/flux1-dev.safetensors"'
-print(f"Executing command: {cmd}")
-os.system(cmd)
+# `git-lfs` で LFS ファイルを取得
+print("Fetching model file using Git LFS...")
+subprocess.run(["git", "lfs", "pull"], cwd=MODEL_DIR, check=True)
 
-# ファイルサイズチェック
-if os.path.exists(model_path):
-    file_size = os.path.getsize(model_path)
+# モデルのサイズを確認
+if os.path.exists(MODEL_PATH):
+    file_size = os.path.getsize(MODEL_PATH)
     if file_size < 1000:  # 1KB 未満なら異常
-        print(f"Error: {model_path} is too small ({file_size} bytes). Download failed.")
-        os.remove(model_path)
+        print(f"Error: {MODEL_PATH} is too small ({file_size} bytes). Download failed.")
+        os.remove(MODEL_PATH)
         exit(1)
     else:
-        print(f"Download successful. File size: {file_size} bytes.")
+        print(f"Download successful. File size: {file_size / (1024 * 1024 * 1024):.2f} GB.")
 else:
-    print("Error: File not found after download.")
+    print("Error: Model file not found after download.")
     exit(1)
 
-
     
 print("Done\nStarting Webui...")
 os.chdir(user_home / r"stable-diffusion-webui")