Upload 3 files

convert_url_to_diffusers_flux_gr.py

@@ -57,9 +57,11 @@ def clear_sd(sd: dict):
     gc.collect()
 
 def clone_sd(sd: dict):
+    from copy import deepcopy
     print("Cloning state dict.")
     for k in list(sd.keys()):
-        sd[k] = sd.pop(k).detach().clone()
+        sd[k] = deepcopy(sd.pop(k))
+        #sd[k] = sd.pop(k).detach().clone().to(device="cpu")
     torch.cuda.empty_cache()
     gc.collect()
 
@@ -384,7 +386,7 @@ def read_safetensors_metadata(path):
         header_json = f.read(header_size).decode('utf-8')
         header = json.loads(header_json)
         metadata = header.get('__metadata__', {})
-        return metadata
+        return metadata.copy()
 
 def normalize_key(k: str):
     return k.replace("vae.", "").replace("model.diffusion_model.", "")\
@@ -438,6 +440,7 @@ with torch.no_grad():
                 progress(0.5, desc=f"Saving FLUX.1 {name} to: {path}")
                 print(f"Saving FLUX.1 {name} to: {path}")
                 if metadata is not None:
+                    progress(0.5, desc=f"Saving FLUX.1 {name} metadata to: {path}")
                     save_torch_state_dict(state_dict=sd, save_directory=path,
                                         filename_pattern=pattern, max_shard_size=size, metadata=metadata)
                 else:

requirements.txt

@@ -1,15 +1,9 @@
-huggingface_hub
+torch
 safetensors
-git+https://github.com/huggingface/transformers
-git+https://github.com/huggingface/accelerate
-git+https://github.com/huggingface/diffusers
-git+https://github.com/huggingface/peft
+git+https://github.com/huggingface/huggingface_hub
 optimum-quanto
 sentencepiece
-torch
-torchaudio
-torchvision
-pytorch_lightning
+git+https://github.com/Lightning-AI/pytorch-lightning
 aria2
 gdown
 gguf>=0.9.1