mirror of
https://github.com/2noise/ChatTTS.git
synced 2026-08-28 17:45:02 +08:00
fix(examples): conda package name conflict (#701)
This commit is contained in:
@@ -1,6 +1,13 @@
|
||||
import os, sys
|
||||
|
||||
if sys.platform == "darwin":
|
||||
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
|
||||
|
||||
now_dir = os.getcwd()
|
||||
sys.path.append(now_dir)
|
||||
|
||||
from dataclasses import asdict
|
||||
import argparse
|
||||
import os
|
||||
import torch
|
||||
from tqdm import tqdm
|
||||
from ChatTTS.model.dvae import DVAE
|
||||
@@ -8,7 +15,8 @@ from ChatTTS.config import Config
|
||||
from vocos import Vocos
|
||||
from vocos.pretrained import instantiate_class
|
||||
import torch.jit as jit
|
||||
from examples.onnx.gpt import GPT
|
||||
|
||||
from gpt import GPT
|
||||
|
||||
# disable cuda
|
||||
torch.cuda.is_available = lambda: False
|
||||
|
||||
@@ -4,7 +4,8 @@ from typing import Tuple
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
from torch.nn.utils.parametrizations import weight_norm
|
||||
from .modeling_llama import LlamaModel, LlamaConfig
|
||||
|
||||
from modeling_llama import LlamaModel, LlamaConfig
|
||||
|
||||
|
||||
class GPT(nn.Module):
|
||||
|
||||
@@ -10,8 +10,8 @@ import argparse
|
||||
|
||||
import gradio as gr
|
||||
|
||||
from examples.web.funcs import *
|
||||
from examples.web.ex import ex
|
||||
from funcs import *
|
||||
from ex import ex
|
||||
|
||||
|
||||
def main():
|
||||
|
||||
Reference in New Issue
Block a user