fantaxy commited on
Commit
67762a6
1 Parent(s): 953b43a

Update app-backup.py

Browse files
Files changed (1) hide show
  1. app-backup.py +4 -8
app-backup.py CHANGED
@@ -16,7 +16,7 @@ from huggingface_hub import snapshot_download
16
  import spaces
17
 
18
  device = "cuda"
19
- root_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
20
  ckpt_dir = f'{root_dir}/weights/Kolors'
21
 
22
  snapshot_download(repo_id="Kwai-Kolors/Kolors", local_dir=ckpt_dir)
@@ -48,9 +48,6 @@ pipe = StableDiffusionXLPipeline(
48
  force_zeros_for_empty_prompt=False
49
  ).to(device)
50
 
51
- #pipe = pipe.to(device)
52
- #pipe.enable_model_cpu_offload()
53
-
54
  if hasattr(pipe.unet, 'encoder_hid_proj'):
55
  pipe.unet.text_encoder_hid_proj = pipe.unet.encoder_hid_proj
56
 
@@ -175,7 +172,7 @@ with gr.Blocks(css=css) as demo:
175
  minimum=1,
176
  maximum=100,
177
  step=1,
178
- value=100,
179
  )
180
 
181
  gr.Examples(
@@ -193,6 +190,5 @@ with gr.Blocks(css=css) as demo:
193
  outputs=[result, seed]
194
  )
195
 
196
- # 포트 7890 설정, 대기열 활성화, API 활성화
197
- demo.launch()
198
-
 
16
  import spaces
17
 
18
  device = "cuda"
19
+ root_dir = os.getcwd()
20
  ckpt_dir = f'{root_dir}/weights/Kolors'
21
 
22
  snapshot_download(repo_id="Kwai-Kolors/Kolors", local_dir=ckpt_dir)
 
48
  force_zeros_for_empty_prompt=False
49
  ).to(device)
50
 
 
 
 
51
  if hasattr(pipe.unet, 'encoder_hid_proj'):
52
  pipe.unet.text_encoder_hid_proj = pipe.unet.encoder_hid_proj
53
 
 
172
  minimum=1,
173
  maximum=100,
174
  step=1,
175
+ value=50,
176
  )
177
 
178
  gr.Examples(
 
190
  outputs=[result, seed]
191
  )
192
 
193
+ # Launch the app
194
+ demo.launch(share=True)