ginipick commited on
Commit
49cebed
·
verified ·
1 Parent(s): a5c1cb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -7,6 +7,9 @@ import os
7
 
8
  import spaces
9
 
 
 
 
10
  from transformers import CLIPVisionModelWithProjection, CLIPImageProcessor
11
  from kolors.pipelines.pipeline_stable_diffusion_xl_chatglm_256_ipadapter import StableDiffusionXLPipeline
12
  from kolors.models.modeling_chatglm import ChatGLMModel
@@ -15,8 +18,6 @@ from kolors.models.unet_2d_condition import UNet2DConditionModel
15
  from diffusers import AutoencoderKL, EulerDiscreteScheduler
16
 
17
 
18
- from huggingface_hub import hf_hub_download, model_info
19
-
20
  device = "cuda"
21
  root_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
22
  ckpt_dir = f'{root_dir}/weights/Kolors'
@@ -330,4 +331,4 @@ with gr.Blocks(theme="apriel", css=css) as demo:
330
  outputs=[result, seed]
331
  )
332
 
333
- demo.queue().launch()
 
7
 
8
  import spaces
9
 
10
+ # 수정: 명시적으로 snapshot_download 임포트 추가
11
+ from huggingface_hub import snapshot_download, hf_hub_download, model_info
12
+
13
  from transformers import CLIPVisionModelWithProjection, CLIPImageProcessor
14
  from kolors.pipelines.pipeline_stable_diffusion_xl_chatglm_256_ipadapter import StableDiffusionXLPipeline
15
  from kolors.models.modeling_chatglm import ChatGLMModel
 
18
  from diffusers import AutoencoderKL, EulerDiscreteScheduler
19
 
20
 
 
 
21
  device = "cuda"
22
  root_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
23
  ckpt_dir = f'{root_dir}/weights/Kolors'
 
331
  outputs=[result, seed]
332
  )
333
 
334
+ demo.queue().launch()