JiantaoLin commited on
Commit
1809f2a
·
1 Parent(s): 6034149
Files changed (2) hide show
  1. app.py +3 -6
  2. pipeline/kiss3d_wrapper.py +0 -1
app.py CHANGED
@@ -80,12 +80,9 @@ LOGO_PATH = "app_assets/logo_temp_.png" # Update this to the actual path of you
80
  ARXIV_LINK = "https://arxiv.org/abs/example"
81
  GITHUB_LINK = "https://github.com/example"
82
 
83
- k3d_wrapper = None
84
- @spaces.GPU
85
- def init_warpper():
86
- global k3d_wrapper
87
- k3d_wrapper = init_wrapper_from_config('./pipeline/pipeline_config/default.yaml')
88
- init_warpper()
89
 
90
  from models.ISOMER.scripts.utils import fix_vert_color_glb
91
  torch.backends.cuda.matmul.allow_tf32 = True
 
80
  ARXIV_LINK = "https://arxiv.org/abs/example"
81
  GITHUB_LINK = "https://github.com/example"
82
 
83
+
84
+ k3d_wrapper = init_wrapper_from_config('./pipeline/pipeline_config/default.yaml')
85
+
 
 
 
86
 
87
  from models.ISOMER.scripts.utils import fix_vert_color_glb
88
  torch.backends.cuda.matmul.allow_tf32 = True
pipeline/kiss3d_wrapper.py CHANGED
@@ -49,7 +49,6 @@ def convert_flux_pipeline(exist_flux_pipe, target_pipe, **kwargs):
49
  )
50
  return new_pipe
51
 
52
- @spaces.GPU
53
  def init_wrapper_from_config(config_path):
54
  with open(config_path, 'r') as config_file:
55
  config_ = yaml.load(config_file, yaml.FullLoader)
 
49
  )
50
  return new_pipe
51
 
 
52
  def init_wrapper_from_config(config_path):
53
  with open(config_path, 'r') as config_file:
54
  config_ = yaml.load(config_file, yaml.FullLoader)