lemonaddie commited on
Commit
eaed46e
·
verified ·
1 Parent(s): 58b59e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- print(device)
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: