Spaces:
Runtime error
Runtime error
lemonaddie
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -504,7 +504,7 @@ def main():
|
|
504 |
from pipeline.depth_normal_pipeline_clip_cfg import DepthNormalEstimationPipeline
|
505 |
|
506 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
507 |
-
|
508 |
pipe = DepthNormalEstimationPipeline.from_pretrained(CHECKPOINT)
|
509 |
|
510 |
try:
|
|
|
504 |
from pipeline.depth_normal_pipeline_clip_cfg import DepthNormalEstimationPipeline
|
505 |
|
506 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
507 |
+
a = torch.zeros((1), device=device)
|
508 |
pipe = DepthNormalEstimationPipeline.from_pretrained(CHECKPOINT)
|
509 |
|
510 |
try:
|