Update convert.py

convert.py

@@ -14,11 +14,18 @@ from huggingface_hub import CommitInfo, Discussion, HfApi, hf_hub_download
 from huggingface_hub.file_download import repo_folder_name
 from diffusers import StableDiffusionXLPipeline
 from transformers import CONFIG_MAPPING
+import spaces
 
 
 COMMIT_MESSAGE = " This PR adds fp32 and fp16 weights in safetensors format to {}"
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
+if torch.cuda.is_available():
+    torch_dtype = torch.bfloat16
+else:
+    torch_dtype = torch.float32
 
+@spaces.GPU
 def convert_single(model_id: str, filename: str, folder: str, progress: Any, token: str):
     progress(0, desc="Downloading model")
     local_file = os.path.join(model_id, filename)