fix(examples): conda package name conflict (#701)

2noise · Aug 23, 2024 · 554554c · 554554c
1 parent a27357f
commit 554554c
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 5 deletions.
diff --git a/examples/onnx/exporter.py b/examples/onnx/exporter.py
@@ -1,14 +1,22 @@
+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
 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

diff --git a/examples/onnx/gpt.py b/examples/onnx/gpt.py
@@ -4,7 +4,8 @@
 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):

diff --git a/examples/web/webui.py b/examples/web/webui.py
@@ -10,8 +10,8 @@
 
 import gradio as gr
 
-from examples.web.funcs import *
-from examples.web.ex import ex
+from funcs import *
+from ex import ex
 
 
 def main():