Update app.py

app.py

@@ -75,21 +75,24 @@ def load_models(model_path, enable_compile=False, attention_backend="native"):
     
     if not os.path.exists(model_path):
         vae = AutoencoderKL.from_pretrained(
-            f"{model_path}/vae",
+            f"{model_path}",
+            subfolder="vae",
             torch_dtype=torch.bfloat16,
             device_map="cuda",
             use_auth_token=use_auth_token
         )
         
         text_encoder = AutoModel.from_pretrained(
-            f"{model_path}/text_encoder",
+            f"{model_path}",
+            subfolder="text_encoder",
             torch_dtype=torch.bfloat16,
             device_map="cuda",
             use_auth_token=use_auth_token
         ).eval()
         
         tokenizer = AutoTokenizer.from_pretrained(
-            f"{model_path}/tokenizer",
+            f"{model_path}",
+            subfolder="tokenizer",
             use_auth_token=use_auth_token
         )
     else: