haowu11 commited on
Commit
eeb427f
·
verified ·
1 Parent(s): 49680b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -67,6 +67,7 @@ face_info_generator = FaceInfoGenerator(root_dir = "./")
67
  clip_image_encoder = CLIPVisionModelWithProjection.from_pretrained("Kwai-Kolors/Kolors-IP-Adapter-FaceID-Plus",cache_dir='./',subfolder="clip-vit-large-patch14-336", ignore_mismatched_sizes=True)
68
  clip_image_processor = CLIPImageProcessor(size = 336, crop_size = 336)
69
  snapshotname = os.listdir('./models--Kwai-Kolors--Kolors-IP-Adapter-FaceID-Plus/snapshots')[0]
 
70
  pipe = StableDiffusionXLControlNetImg2ImgPipeline(
71
  vae=vae,
72
  controlnet = controlnet,
 
67
  clip_image_encoder = CLIPVisionModelWithProjection.from_pretrained("Kwai-Kolors/Kolors-IP-Adapter-FaceID-Plus",cache_dir='./',subfolder="clip-vit-large-patch14-336", ignore_mismatched_sizes=True)
68
  clip_image_processor = CLIPImageProcessor(size = 336, crop_size = 336)
69
  snapshotname = os.listdir('./models--Kwai-Kolors--Kolors-IP-Adapter-FaceID-Plus/snapshots')[0]
70
+ print(os.listdir(f'models--Kwai-Kolors--Kolors-IP-Adapter-FaceID-Plus/snapshots/{snapshotname}))
71
  pipe = StableDiffusionXLControlNetImg2ImgPipeline(
72
  vae=vae,
73
  controlnet = controlnet,