Upload app.py

app.py

@@ -14,6 +14,7 @@ from torchvision import transforms
 from dataclasses import dataclass
 import math
 from typing import Callable
+from pathlib import Path
 
 import torch
 import random
@@ -641,6 +642,7 @@ with EmptyInitWrapper():
     model = Flux().to(dtype=torch.bfloat16, device=device)
     
     sd = load_file(f"{model_path}/consolidated_s6700.safetensors")
+    Path(f"{model_path}/consolidated_s6700.safetensors").unlink()
     sd = {k.replace("model.", ""): v for k, v in sd.items()}
     result = model.load_state_dict(sd)