├── .flake8 ├── .github └── workflows │ ├── lint.yaml │ ├── release.yaml │ └── unit_test_cpu.yaml ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── assets ├── LlamaForASR.png ├── Touchnet_16_9.jpg ├── Touchnet_1_1.jpg ├── audio_clip.png ├── format.jpg ├── perf_data.png ├── timewaste.png ├── touchdataset.png ├── touchnet_pretrain.png └── version.txt ├── docs ├── TouchAudioForCausalLM.md ├── audio_pretrain.md ├── audio_sft_asr.md ├── data.md └── text_pretrain.md ├── examples ├── audio │ ├── pretrain │ │ ├── README.md │ │ ├── emilia │ │ │ └── path.sh │ │ └── wenetspeech │ │ │ ├── parse_options.sh │ │ │ ├── path.sh │ │ │ ├── run.sh │ │ │ └── touchnet │ └── sft │ │ └── asr │ │ ├── README.md │ │ └── wenetspeech │ │ ├── config │ │ ├── Kimi-Audio-7B-Instruct.json │ │ ├── Kimi-Audio-7B.json │ │ └── Qwen2-Audio-7B.json │ │ ├── local │ │ └── extract_trans_and_pred.py │ │ ├── parse_options.sh │ │ ├── path.sh │ │ ├── run.sh │ │ └── touchnet └── text │ └── pretrain │ ├── README.md │ ├── allenai_c4 │ ├── config │ │ └── Llama-3_2-1B.json │ ├── download_c4.py │ ├── parse_options.sh │ ├── path.sh │ ├── run.sh │ └── touchnet │ └── fineweb-edu │ ├── config │ └── Llama-3_2-1B.json │ ├── download_fineweb-edu.py │ ├── parse_options.sh │ ├── path.sh │ ├── run.sh │ └── touchnet ├── install_cuda_cudnn.sh ├── pyproject.toml ├── tests ├── assets │ ├── config │ │ └── tiny_llama.json │ └── dataset │ │ ├── aishell-BAC009S0724W0121.wav │ │ ├── data.jsonl │ │ └── librispeech-1995-1837-0001.wav └── touchnet │ ├── bin │ └── test_make_data.py │ ├── data │ └── test_dataloader.py │ ├── models │ └── test_llama.py │ └── utils │ ├── distributed_cpu.py │ ├── test_distributed_cpu.py │ └── test_pack_loss.py └── touchnet ├── __init__.py ├── bin ├── __init__.py ├── convert_dcp_to_hf.py ├── convert_hf_to_dcp.py ├── error_rate_zh ├── make_data.py ├── textnorm_zh.py └── train.py ├── data ├── __init__.py ├── dataloader.py ├── datapipe.py ├── dataset.py └── functions.py ├── loss ├── __init__.py └── cross_entropy.py ├── models ├── __init__.py ├── helper_func.py ├── kimi_audio │ ├── __init__.py │ ├── configuration_kimi_audio.py │ ├── inference_kimi_audio.py │ ├── modeling_kimi_audio.py │ ├── parallelize_kimi_audio.py │ └── processing_kimi_audio.py ├── llama │ ├── __init__.py │ ├── parallelize_llama.py │ ├── pipeline_llama.py │ └── processing_llama.py ├── qwen2_audio │ ├── __init__.py │ ├── inference_qwen2_audio.py │ ├── parallelize_qwen2_audio.py │ └── processing_qwen2_audio.py └── touch_audio │ ├── __init__.py │ ├── configuration_touch_audio.py │ ├── inference_touch_audio.py │ ├── modeling_touch_audio.py │ ├── parallelize_touch_audio.py │ └── processing_touch_audio.py ├── tokenizer ├── __init__.py └── tokenizer.py └── utils ├── __init__.py ├── checkpoint.py ├── distributed.py ├── inference.py ├── logging.py ├── metrics.py ├── optimizer.py ├── profiling.py └── train_spec.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/.flake8 -------------------------------------------------------------------------------- /.github/workflows/lint.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/.github/workflows/lint.yaml -------------------------------------------------------------------------------- /.github/workflows/release.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/.github/workflows/release.yaml -------------------------------------------------------------------------------- /.github/workflows/unit_test_cpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/.github/workflows/unit_test_cpu.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/README.md -------------------------------------------------------------------------------- /assets/LlamaForASR.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/assets/LlamaForASR.png -------------------------------------------------------------------------------- /assets/Touchnet_16_9.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/assets/Touchnet_16_9.jpg -------------------------------------------------------------------------------- /assets/Touchnet_1_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/assets/Touchnet_1_1.jpg -------------------------------------------------------------------------------- /assets/audio_clip.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/assets/audio_clip.png -------------------------------------------------------------------------------- /assets/format.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/assets/format.jpg -------------------------------------------------------------------------------- /assets/perf_data.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/assets/perf_data.png -------------------------------------------------------------------------------- /assets/timewaste.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/assets/timewaste.png -------------------------------------------------------------------------------- /assets/touchdataset.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/assets/touchdataset.png -------------------------------------------------------------------------------- /assets/touchnet_pretrain.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/assets/touchnet_pretrain.png -------------------------------------------------------------------------------- /assets/version.txt: -------------------------------------------------------------------------------- 1 | 0.1.0 2 | -------------------------------------------------------------------------------- /docs/TouchAudioForCausalLM.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/docs/TouchAudioForCausalLM.md -------------------------------------------------------------------------------- /docs/audio_pretrain.md: -------------------------------------------------------------------------------- 1 | ../examples/audio/pretrain/README.md -------------------------------------------------------------------------------- /docs/audio_sft_asr.md: -------------------------------------------------------------------------------- 1 | ../examples/audio/sft/asr/README.md -------------------------------------------------------------------------------- /docs/data.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/docs/data.md -------------------------------------------------------------------------------- /docs/text_pretrain.md: -------------------------------------------------------------------------------- 1 | ../examples/text/pretrain/README.md -------------------------------------------------------------------------------- /examples/audio/pretrain/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/audio/pretrain/README.md -------------------------------------------------------------------------------- /examples/audio/pretrain/emilia/path.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/audio/pretrain/emilia/path.sh -------------------------------------------------------------------------------- /examples/audio/pretrain/wenetspeech/parse_options.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/audio/pretrain/wenetspeech/parse_options.sh -------------------------------------------------------------------------------- /examples/audio/pretrain/wenetspeech/path.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/audio/pretrain/wenetspeech/path.sh -------------------------------------------------------------------------------- /examples/audio/pretrain/wenetspeech/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/audio/pretrain/wenetspeech/run.sh -------------------------------------------------------------------------------- /examples/audio/pretrain/wenetspeech/touchnet: -------------------------------------------------------------------------------- 1 | ../../../../touchnet -------------------------------------------------------------------------------- /examples/audio/sft/asr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/audio/sft/asr/README.md -------------------------------------------------------------------------------- /examples/audio/sft/asr/wenetspeech/config/Kimi-Audio-7B-Instruct.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/audio/sft/asr/wenetspeech/config/Kimi-Audio-7B-Instruct.json -------------------------------------------------------------------------------- /examples/audio/sft/asr/wenetspeech/config/Kimi-Audio-7B.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/audio/sft/asr/wenetspeech/config/Kimi-Audio-7B.json -------------------------------------------------------------------------------- /examples/audio/sft/asr/wenetspeech/config/Qwen2-Audio-7B.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/audio/sft/asr/wenetspeech/config/Qwen2-Audio-7B.json -------------------------------------------------------------------------------- /examples/audio/sft/asr/wenetspeech/local/extract_trans_and_pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/audio/sft/asr/wenetspeech/local/extract_trans_and_pred.py -------------------------------------------------------------------------------- /examples/audio/sft/asr/wenetspeech/parse_options.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/audio/sft/asr/wenetspeech/parse_options.sh -------------------------------------------------------------------------------- /examples/audio/sft/asr/wenetspeech/path.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/audio/sft/asr/wenetspeech/path.sh -------------------------------------------------------------------------------- /examples/audio/sft/asr/wenetspeech/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/audio/sft/asr/wenetspeech/run.sh -------------------------------------------------------------------------------- /examples/audio/sft/asr/wenetspeech/touchnet: -------------------------------------------------------------------------------- 1 | ../../../../../touchnet -------------------------------------------------------------------------------- /examples/text/pretrain/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/text/pretrain/README.md -------------------------------------------------------------------------------- /examples/text/pretrain/allenai_c4/config/Llama-3_2-1B.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/text/pretrain/allenai_c4/config/Llama-3_2-1B.json -------------------------------------------------------------------------------- /examples/text/pretrain/allenai_c4/download_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/text/pretrain/allenai_c4/download_c4.py -------------------------------------------------------------------------------- /examples/text/pretrain/allenai_c4/parse_options.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/text/pretrain/allenai_c4/parse_options.sh -------------------------------------------------------------------------------- /examples/text/pretrain/allenai_c4/path.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/text/pretrain/allenai_c4/path.sh -------------------------------------------------------------------------------- /examples/text/pretrain/allenai_c4/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/text/pretrain/allenai_c4/run.sh -------------------------------------------------------------------------------- /examples/text/pretrain/allenai_c4/touchnet: -------------------------------------------------------------------------------- 1 | ../../../../touchnet -------------------------------------------------------------------------------- /examples/text/pretrain/fineweb-edu/config/Llama-3_2-1B.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/text/pretrain/fineweb-edu/config/Llama-3_2-1B.json -------------------------------------------------------------------------------- /examples/text/pretrain/fineweb-edu/download_fineweb-edu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/text/pretrain/fineweb-edu/download_fineweb-edu.py -------------------------------------------------------------------------------- /examples/text/pretrain/fineweb-edu/parse_options.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/text/pretrain/fineweb-edu/parse_options.sh -------------------------------------------------------------------------------- /examples/text/pretrain/fineweb-edu/path.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/text/pretrain/fineweb-edu/path.sh -------------------------------------------------------------------------------- /examples/text/pretrain/fineweb-edu/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/examples/text/pretrain/fineweb-edu/run.sh -------------------------------------------------------------------------------- /examples/text/pretrain/fineweb-edu/touchnet: -------------------------------------------------------------------------------- 1 | ../../../../touchnet -------------------------------------------------------------------------------- /install_cuda_cudnn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/install_cuda_cudnn.sh -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/pyproject.toml -------------------------------------------------------------------------------- /tests/assets/config/tiny_llama.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/tests/assets/config/tiny_llama.json -------------------------------------------------------------------------------- /tests/assets/dataset/aishell-BAC009S0724W0121.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/tests/assets/dataset/aishell-BAC009S0724W0121.wav -------------------------------------------------------------------------------- /tests/assets/dataset/data.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/tests/assets/dataset/data.jsonl -------------------------------------------------------------------------------- /tests/assets/dataset/librispeech-1995-1837-0001.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/tests/assets/dataset/librispeech-1995-1837-0001.wav -------------------------------------------------------------------------------- /tests/touchnet/bin/test_make_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/tests/touchnet/bin/test_make_data.py -------------------------------------------------------------------------------- /tests/touchnet/data/test_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/tests/touchnet/data/test_dataloader.py -------------------------------------------------------------------------------- /tests/touchnet/models/test_llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/tests/touchnet/models/test_llama.py -------------------------------------------------------------------------------- /tests/touchnet/utils/distributed_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/tests/touchnet/utils/distributed_cpu.py -------------------------------------------------------------------------------- /tests/touchnet/utils/test_distributed_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/tests/touchnet/utils/test_distributed_cpu.py -------------------------------------------------------------------------------- /tests/touchnet/utils/test_pack_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/tests/touchnet/utils/test_pack_loss.py -------------------------------------------------------------------------------- /touchnet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/__init__.py -------------------------------------------------------------------------------- /touchnet/bin/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/bin/__init__.py -------------------------------------------------------------------------------- /touchnet/bin/convert_dcp_to_hf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/bin/convert_dcp_to_hf.py -------------------------------------------------------------------------------- /touchnet/bin/convert_hf_to_dcp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/bin/convert_hf_to_dcp.py -------------------------------------------------------------------------------- /touchnet/bin/error_rate_zh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/bin/error_rate_zh -------------------------------------------------------------------------------- /touchnet/bin/make_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/bin/make_data.py -------------------------------------------------------------------------------- /touchnet/bin/textnorm_zh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/bin/textnorm_zh.py -------------------------------------------------------------------------------- /touchnet/bin/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/bin/train.py -------------------------------------------------------------------------------- /touchnet/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/data/__init__.py -------------------------------------------------------------------------------- /touchnet/data/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/data/dataloader.py -------------------------------------------------------------------------------- /touchnet/data/datapipe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/data/datapipe.py -------------------------------------------------------------------------------- /touchnet/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/data/dataset.py -------------------------------------------------------------------------------- /touchnet/data/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/data/functions.py -------------------------------------------------------------------------------- /touchnet/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/loss/__init__.py -------------------------------------------------------------------------------- /touchnet/loss/cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/loss/cross_entropy.py -------------------------------------------------------------------------------- /touchnet/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/__init__.py -------------------------------------------------------------------------------- /touchnet/models/helper_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/helper_func.py -------------------------------------------------------------------------------- /touchnet/models/kimi_audio/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/kimi_audio/__init__.py -------------------------------------------------------------------------------- /touchnet/models/kimi_audio/configuration_kimi_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/kimi_audio/configuration_kimi_audio.py -------------------------------------------------------------------------------- /touchnet/models/kimi_audio/inference_kimi_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/kimi_audio/inference_kimi_audio.py -------------------------------------------------------------------------------- /touchnet/models/kimi_audio/modeling_kimi_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/kimi_audio/modeling_kimi_audio.py -------------------------------------------------------------------------------- /touchnet/models/kimi_audio/parallelize_kimi_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/kimi_audio/parallelize_kimi_audio.py -------------------------------------------------------------------------------- /touchnet/models/kimi_audio/processing_kimi_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/kimi_audio/processing_kimi_audio.py -------------------------------------------------------------------------------- /touchnet/models/llama/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/llama/__init__.py -------------------------------------------------------------------------------- /touchnet/models/llama/parallelize_llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/llama/parallelize_llama.py -------------------------------------------------------------------------------- /touchnet/models/llama/pipeline_llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/llama/pipeline_llama.py -------------------------------------------------------------------------------- /touchnet/models/llama/processing_llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/llama/processing_llama.py -------------------------------------------------------------------------------- /touchnet/models/qwen2_audio/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/qwen2_audio/__init__.py -------------------------------------------------------------------------------- /touchnet/models/qwen2_audio/inference_qwen2_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/qwen2_audio/inference_qwen2_audio.py -------------------------------------------------------------------------------- /touchnet/models/qwen2_audio/parallelize_qwen2_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/qwen2_audio/parallelize_qwen2_audio.py -------------------------------------------------------------------------------- /touchnet/models/qwen2_audio/processing_qwen2_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/qwen2_audio/processing_qwen2_audio.py -------------------------------------------------------------------------------- /touchnet/models/touch_audio/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/touch_audio/__init__.py -------------------------------------------------------------------------------- /touchnet/models/touch_audio/configuration_touch_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/touch_audio/configuration_touch_audio.py -------------------------------------------------------------------------------- /touchnet/models/touch_audio/inference_touch_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/touch_audio/inference_touch_audio.py -------------------------------------------------------------------------------- /touchnet/models/touch_audio/modeling_touch_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/touch_audio/modeling_touch_audio.py -------------------------------------------------------------------------------- /touchnet/models/touch_audio/parallelize_touch_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/touch_audio/parallelize_touch_audio.py -------------------------------------------------------------------------------- /touchnet/models/touch_audio/processing_touch_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/models/touch_audio/processing_touch_audio.py -------------------------------------------------------------------------------- /touchnet/tokenizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/tokenizer/__init__.py -------------------------------------------------------------------------------- /touchnet/tokenizer/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/tokenizer/tokenizer.py -------------------------------------------------------------------------------- /touchnet/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/utils/__init__.py -------------------------------------------------------------------------------- /touchnet/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/utils/checkpoint.py -------------------------------------------------------------------------------- /touchnet/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/utils/distributed.py -------------------------------------------------------------------------------- /touchnet/utils/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/utils/inference.py -------------------------------------------------------------------------------- /touchnet/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/utils/logging.py -------------------------------------------------------------------------------- /touchnet/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/utils/metrics.py -------------------------------------------------------------------------------- /touchnet/utils/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/utils/optimizer.py -------------------------------------------------------------------------------- /touchnet/utils/profiling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/utils/profiling.py -------------------------------------------------------------------------------- /touchnet/utils/train_spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingchensong/TouchNet/HEAD/touchnet/utils/train_spec.py --------------------------------------------------------------------------------