lemonaddie commited on
Commit
a738d23
·
verified ·
1 Parent(s): 21c7fd8

Update app2.py

Browse files
Files changed (1) hide show
  1. app2.py +2 -1
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,