lemonaddie commited on
Commit
0e4b3d5
1 Parent(s): 01a7879

Update app2.py

Browse files
Files changed (1) hide show
  1. app2.py +1 -1
app2.py CHANGED
@@ -53,7 +53,7 @@ sd_image_variations_diffusers_path = '.'
53
  image_encoder = CLIPVisionModelWithProjection.from_pretrained(sd_image_variations_diffusers_path, subfolder="image_encoder")
54
  feature_extractor = CLIPImageProcessor.from_pretrained(sd_image_variations_diffusers_path, subfolder="feature_extractor")
55
 
56
- unet = UNet2DConditionModel.from_pretrained('tbd')
57
 
58
  pipe = DepthNormalEstimationPipeline(vae=vae,
59
  image_encoder=image_encoder,
 
53
  image_encoder = CLIPVisionModelWithProjection.from_pretrained(sd_image_variations_diffusers_path, subfolder="image_encoder")
54
  feature_extractor = CLIPImageProcessor.from_pretrained(sd_image_variations_diffusers_path, subfolder="feature_extractor")
55
 
56
+ unet = UNet2DConditionModel.from_pretrained('./wocfg/unet_ema')
57
 
58
  pipe = DepthNormalEstimationPipeline(vae=vae,
59
  image_encoder=image_encoder,