JiantaoLin
commited on
Commit
·
0d6adf0
1
Parent(s):
da89ea5
new
Browse files- app.py +1 -1
- 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 |
-
|
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)
|