JiantaoLin commited on
Commit
0d6adf0
·
1 Parent(s): da89ea5
Files changed (2) hide show
  1. app.py +1 -1
  2. pipeline/kiss3d_wrapper.py +1 -1
app.py CHANGED
@@ -142,7 +142,7 @@ def save_py3dmesh_with_trimesh_fast(meshes, save_glb_path=TEMP_MESH_ADDRESS, app
142
  #
143
  @spaces.GPU
144
  def text_to_detailed(prompt, seed=None):
145
- # print(torch.cuda.is_available())
146
  # print(f"Before text_to_detailed: {torch.cuda.memory_allocated() / 1024**3} GB")
147
  return k3d_wrapper.get_detailed_prompt(prompt, seed)
148
 
 
142
  #
143
  @spaces.GPU
144
  def text_to_detailed(prompt, seed=None):
145
+ print(f"torch.cuda.is_available():{torch.cuda.is_available()}")
146
  # print(f"Before text_to_detailed: {torch.cuda.memory_allocated() / 1024**3} GB")
147
  return k3d_wrapper.get_detailed_prompt(prompt, seed)
148
 
pipeline/kiss3d_wrapper.py CHANGED
@@ -258,7 +258,7 @@ class kiss3d_wrapper(object):
258
  caption_text = self.get_detailed_prompt(caption_text)
259
 
260
  return caption_text
261
- @spaces.GPU
262
  def get_detailed_prompt(self, prompt, seed=None):
263
  if self.llm_model is not None:
264
  detailed_prompt = get_llm_response(self.llm_model, self.llm_tokenizer, prompt, seed=seed)
 
258
  caption_text = self.get_detailed_prompt(caption_text)
259
 
260
  return caption_text
261
+ # @spaces.GPU
262
  def get_detailed_prompt(self, prompt, seed=None):
263
  if self.llm_model is not None:
264
  detailed_prompt = get_llm_response(self.llm_model, self.llm_tokenizer, prompt, seed=seed)