Spaces:
Runtime error
Runtime error
lemonaddie
commited on
Update app2.py
Browse files
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 |
-
|
69 |
-
|
70 |
-
|
71 |
-
|
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
|