Spaces:
Runtime error
Runtime error
lemonaddie
commited on
Update app2.py
Browse files
app2.py
CHANGED
@@ -54,7 +54,8 @@ sd_image_variations_diffusers_path = '.'
|
|
54 |
image_encoder = CLIPVisionModelWithProjection.from_pretrained(sd_image_variations_diffusers_path, subfolder="image_encoder")
|
55 |
feature_extractor = CLIPImageProcessor.from_pretrained(sd_image_variations_diffusers_path, subfolder="feature_extractor")
|
56 |
|
57 |
-
unet = UNet2DConditionModel.from_pretrained('./wocfg/unet_ema')
|
|
|
58 |
|
59 |
pipe = DepthNormalEstimationPipeline(vae=vae,
|
60 |
image_encoder=image_encoder,
|
|
|
54 |
image_encoder = CLIPVisionModelWithProjection.from_pretrained(sd_image_variations_diffusers_path, subfolder="image_encoder")
|
55 |
feature_extractor = CLIPImageProcessor.from_pretrained(sd_image_variations_diffusers_path, subfolder="feature_extractor")
|
56 |
|
57 |
+
#unet = UNet2DConditionModel.from_pretrained('./wocfg/unet_ema')
|
58 |
+
unet = UNet2DConditionModel.from_pretrained('./cfg/unet_ema')
|
59 |
|
60 |
pipe = DepthNormalEstimationPipeline(vae=vae,
|
61 |
image_encoder=image_encoder,
|