Update app.py
Browse files
app.py
CHANGED
|
@@ -383,7 +383,7 @@ def main():
|
|
| 383 |
from pipeline.depth_normal_pipeline_clip_cfg import DepthNormalEstimationPipeline
|
| 384 |
|
| 385 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 386 |
-
pipe = DepthNormalEstimationPipeline.from_pretrained(CHECKPOINT)
|
| 387 |
|
| 388 |
try:
|
| 389 |
import xformers
|
|
|
|
| 383 |
from pipeline.depth_normal_pipeline_clip_cfg import DepthNormalEstimationPipeline
|
| 384 |
|
| 385 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 386 |
+
pipe = DepthNormalEstimationPipeline.from_pretrained(CHECKPOINT, use_safetensors=False)
|
| 387 |
|
| 388 |
try:
|
| 389 |
import xformers
|