haowu11 commited on
Commit
7787e71
·
verified ·
1 Parent(s): 8aacc4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import torch
 
2
  from transformers import CLIPVisionModelWithProjection,CLIPImageProcessor
3
  from diffusers.utils import load_image
4
  import os,sys
@@ -65,7 +66,7 @@ face_info_generator = FaceInfoGenerator(root_dir = "./")
65
 
66
  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)
67
  clip_image_processor = CLIPImageProcessor(size = 336, crop_size = 336)
68
-
69
  pipe = StableDiffusionXLControlNetImg2ImgPipeline(
70
  vae=vae,
71
  controlnet = controlnet,
 
1
  import torch
2
+ import spaces
3
  from transformers import CLIPVisionModelWithProjection,CLIPImageProcessor
4
  from diffusers.utils import load_image
5
  import os,sys
 
66
 
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
+ print(os.listdir('./'))
70
  pipe = StableDiffusionXLControlNetImg2ImgPipeline(
71
  vae=vae,
72
  controlnet = controlnet,