Spaces:
Running
on
Zero
Running
on
Zero
xinjie.wang
commited on
Commit
·
c2edd8f
1
Parent(s):
37e4080
update
Browse files
asset3d_gen/models/delight_model.py
CHANGED
|
@@ -58,6 +58,7 @@ class DelightingModel(object):
|
|
| 58 |
pipeline.to(self.device, torch.float16)
|
| 59 |
# pipeline.enable_model_cpu_offload()
|
| 60 |
# pipeline.enable_xformers_memory_efficient_attention()
|
|
|
|
| 61 |
self.pipeline = pipeline
|
| 62 |
|
| 63 |
def recenter_image(
|
|
|
|
| 58 |
pipeline.to(self.device, torch.float16)
|
| 59 |
# pipeline.enable_model_cpu_offload()
|
| 60 |
# pipeline.enable_xformers_memory_efficient_attention()
|
| 61 |
+
|
| 62 |
self.pipeline = pipeline
|
| 63 |
|
| 64 |
def recenter_image(
|