Update app.py

app.py

@@ -15,6 +15,8 @@ from torch import Tensor, nn
 from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5Tokenizer
 from safetensors.torch import load_file
 dtype = torch.bfloat16
+from huggingface_hub import snapshot_download
+print(snapshot_download(repo_id="nyanko7/flux-dev-de-distill"))
 device = "cuda" if torch.cuda.is_available() else "cpu"
 # ---------------- Encoders ----------------
 
@@ -628,6 +630,7 @@ class EmptyInitWrapper(torch.overrides.TorchFunctionMode):
 
 with EmptyInitWrapper():
     model = Flux().to(dtype=torch.bfloat16, device="cuda")
+    
     sd = load_file("./consolidated_s6700.safetensors")
     sd = {k.replace("model.", ""): v for k, v in sd.items()}
     result = model.load_state_dict(sd)