Spaces:
Running
on
Zero
Running
on
Zero
lemonaddie
commited on
Commit
•
49fe562
1
Parent(s):
e88bc8b
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
|