bytedance / bytedance/MegaTTS3

Docker install for Mac M1

Open
#44 4 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
6.1k
Forks
474
PR merge metrics
No merged PRs in 30d

Description

```
WARNING: The requested image's platform (linux/amd64) does not match the detected host platform (linux/arm64/v8) and no specific platform was requested
Start open workers
Running on local URL: http://0.0.0.0:7929

To create a public link, set `share=True` in `launch()`.
IMPORTANT: You are using gradio version 4.19.1, however version 4.44.1 is available, please upgrade.
--------
Process Process-1:
Traceback (most recent call last):
File "/opt/conda/lib/python3.10/multiprocessing/process.py", line 314, in _bootstrap
self.run()
File "/opt/conda/lib/python3.10/multiprocessing/process.py", line 108, in run
self._target(*self._args, **self._kwargs)
File "/app/tts/gradio_api.py", line 28, in model_worker
infer_pipe = MegaTTS3DiTInfer(device=device)
File "/app/tts/infer_cli.py", line 90, in __init__
self.build_model(self.device)
File "/app/tts/infer_cli.py", line 118, in build_model
load_ckpt(self.dur_model, f'{self.dur_exp_name}', 'dur_model')
File "/app/tts/utils/commons/ckpt_utils.py", line 87, in load_ckpt
checkpoint = torch_load_dist(ckpt_path, map_location='cpu')
File "/app/tts/utils/commons/ckpt_utils.py", line 50, in torch_load_dist
checkpoint = torch.load(tmp_path, map_location=map_location)
File "/opt/conda/lib/python3.10/site-packages/torch/serialization.py", line 1040, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "/opt/conda/lib/python3.10/site-packages/torch/serialization.py", line 1262, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, 'v'.
```

Contributor guide

No contributing guide indexed for this repository

Research direction

Reproduce the Docker install on a Mac M1, then start with the platform warning and the traceback through /app/tts/gradio_api.py, /app/tts/infer_cli.py, and /app/tts/utils/commons/ckpt_utils.py. Determine why checkpoint loading fails and document or fix the installation so the worker starts without the reported error.

Written by the indexing model from the issue text.

Assessment

Tech stack
docker, python, pytorch
Domain
devops, machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
30/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.