rocketmandrey commited on
Commit
90b7725
·
verified ·
1 Parent(s): 47d15fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -10,9 +10,11 @@ HF_TOKEN = os.environ["HF_HUB_TOKEN"]
10
  # 2) Загружаем pipeline из репозитория MeiGen-MultiTalk
11
  pipe = DiffusionPipeline.from_pretrained(
12
  "MeiGen-AI/MeiGen-MultiTalk",
13
- use_auth_token=HF_TOKEN,
 
14
  torch_dtype=torch.float16
15
  )
 
16
  pipe = pipe.to("cuda") if torch.cuda.is_available() else pipe.to("cpu")
17
 
18
  # 3) Функция для генерации аудио
 
10
  # 2) Загружаем pipeline из репозитория MeiGen-MultiTalk
11
  pipe = DiffusionPipeline.from_pretrained(
12
  "MeiGen-AI/MeiGen-MultiTalk",
13
+ subfolder="GFengG", # <— обращаемся в подпапку, где лежат *.index.json и веса
14
+ use_auth_token=HF_HUB_TOKEN,
15
  torch_dtype=torch.float16
16
  )
17
+
18
  pipe = pipe.to("cuda") if torch.cuda.is_available() else pipe.to("cpu")
19
 
20
  # 3) Функция для генерации аудио