lemonaddie commited on
Commit
064b3ee
1 Parent(s): dbbe505

Update app2.py

Browse files
Files changed (1) hide show
  1. app2.py +7 -7
app2.py CHANGED
@@ -31,7 +31,7 @@ import sys
31
  sys.path.append("../")
32
  from models.depth_normal_pipeline_clip import DepthNormalEstimationPipeline
33
  #from models.depth_normal_pipeline_clip_cfg import DepthNormalEstimationPipeline
34
- from models.depth_normal_pipeline_clip_cfg_1 import DepthNormalEstimationPipeline as DepthNormalEstimationPipelineCFG
35
  from utils.seed_all import seed_all
36
  import matplotlib.pyplot as plt
37
  from utils.de_normalized import align_scale_shift
@@ -64,11 +64,11 @@ pipe = DepthNormalEstimationPipeline(vae=vae,
64
  unet=unet,
65
  scheduler=scheduler)
66
 
67
- pipe_cfg = DepthNormalEstimationPipelineCFG(vae=vae,
68
- image_encoder=image_encoder,
69
- feature_extractor=feature_extractor,
70
- unet=unet_cfg,
71
- scheduler=scheduler)
72
 
73
 
74
  try:
@@ -78,7 +78,7 @@ except:
78
  pass # run without xformers
79
 
80
  pipe = pipe.to(device)
81
- pipe_cfg = pipe_cfg.to(device)
82
  #run_demo_server(pipe)
83
 
84
  @spaces.GPU
 
31
  sys.path.append("../")
32
  from models.depth_normal_pipeline_clip import DepthNormalEstimationPipeline
33
  #from models.depth_normal_pipeline_clip_cfg import DepthNormalEstimationPipeline
34
+ #from models.depth_normal_pipeline_clip_cfg_1 import DepthNormalEstimationPipeline as DepthNormalEstimationPipelineCFG
35
  from utils.seed_all import seed_all
36
  import matplotlib.pyplot as plt
37
  from utils.de_normalized import align_scale_shift
 
64
  unet=unet,
65
  scheduler=scheduler)
66
 
67
+ # pipe_cfg = DepthNormalEstimationPipelineCFG(vae=vae,
68
+ # image_encoder=image_encoder,
69
+ # feature_extractor=feature_extractor,
70
+ # unet=unet_cfg,
71
+ # scheduler=scheduler)
72
 
73
 
74
  try:
 
78
  pass # run without xformers
79
 
80
  pipe = pipe.to(device)
81
+ #pipe_cfg = pipe_cfg.to(device)
82
  #run_demo_server(pipe)
83
 
84
  @spaces.GPU