Spaces:
Running
on
Zero
Running
on
Zero
xinjie.wang
commited on
Commit
·
82fcd2e
1
Parent(s):
365ef5d
update
Browse files
asset3d_gen/models/text_model.py
CHANGED
|
@@ -107,7 +107,7 @@ def build_text2img_pipeline(
|
|
| 107 |
)
|
| 108 |
pipe = pipe.to(device)
|
| 109 |
# pipe.enable_model_cpu_offload()
|
| 110 |
-
pipe.enable_xformers_memory_efficient_attention()
|
| 111 |
|
| 112 |
return pipe
|
| 113 |
|
|
|
|
| 107 |
)
|
| 108 |
pipe = pipe.to(device)
|
| 109 |
# pipe.enable_model_cpu_offload()
|
| 110 |
+
# pipe.enable_xformers_memory_efficient_attention()
|
| 111 |
|
| 112 |
return pipe
|
| 113 |
|