test

app.py

@@ -5,6 +5,7 @@ import cv2
 import insightface
 import gradio as gr
 import numpy as np
+import os
 from huggingface_hub import snapshot_download
 from transformers import CLIPVisionModelWithProjection,CLIPImageProcessor
 from kolors.pipelines.pipeline_stable_diffusion_xl_chatglm_256_ipadapter_FaceID import StableDiffusionXLPipeline
@@ -30,7 +31,6 @@ unet = UNet2DConditionModel.from_pretrained(f"{ckpt_dir}/unet", revision=None).h
 clip_image_encoder = CLIPVisionModelWithProjection.from_pretrained(f'{ckpt_dir_faceid}/clip-vit-large-patch14-336', ignore_mismatched_sizes=True)
 clip_image_encoder.to(device)
 clip_image_processor = CLIPImageProcessor(size = 336, crop_size = 336)
-print(os.environ['test'])
 
 pipe = StableDiffusionXLPipeline(
     vae = vae,
@@ -84,6 +84,7 @@ def infer(prompt,
           guidance_scale = 5.0, 
           num_inference_steps = 50
         ):
+    print(os.environ['test'])
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed)