From b3f746126c9d32c7ca626deaeac86ae3fb94ddb6 Mon Sep 17 00:00:00 2001
From: ver217 <lhx0217@gmail.com>
Date: Fri, 21 Jun 2024 10:39:18 +0800
Subject: [PATCH] [gradio] hotfix stdit model init

---
 gradio/app.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gradio/app.py b/gradio/app.py
index b1906c0..da5b188 100644
--- a/gradio/app.py
+++ b/gradio/app.py
@@ -100,7 +100,7 @@ def build_models(model_type, config, enable_optimization=False):
     # handle model download logic in HuggingFace Space
     from opensora.models.stdit.stdit3 import STDiT3
 
-    model_kwargs = {k: v for k, v in config.model.items() if k not in ("type", "from_pretrained")}
+    model_kwargs = {k: v for k, v in config.model.items() if k not in ("type", "from_pretrained", "force_huggingface")}
     stdit = STDiT3.from_pretrained(HF_STDIT_MAP[model_type], **model_kwargs)
     stdit = stdit.cuda()