Update app.py

app.py

@@ -14,12 +14,7 @@ from transformers import (
     T5EncoderModel,
     T5Tokenizer
 )
-from accelerate import (
-    init_empty_weights,
-    set_module_tensor_to_device,
-    infer_auto_device_map,
-    load_checkpoint_and_dispatch
-)
+from accelerate import init_empty_weights
 from safetensors import safe_open
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -45,7 +40,7 @@ tokenizer_3 = T5Tokenizer.from_pretrained(model_repo_id, subfolder="tokenizer_3"
 config_file = hf_hub_download(repo_id=model_repo_id, filename="transformer/config.json")
 with open(config_file, "r") as fp:
     config = json.loads(fp)
-with no_init_weights():
+with init_empty_weights():
     transformer = SD3Transformer2DModel.from_config(config)
 
 # Get transformer state dict and load