fountai commited on
Commit
a43c8d9
·
1 Parent(s): 606e2d0
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,5 +1,7 @@
1
  import torch.multiprocessing as mp
2
  import multiprocessing as mp2
 
 
3
  mp.set_start_method('spawn', force=True)
4
  mp2.set_start_method('spawn', force=True)
5
 
@@ -9,6 +11,7 @@ from PIL import Image
9
  import os
10
 
11
  import random
 
12
 
13
  @spaces.GPU(duration=300)
14
  def process_image(image, prompt, steps, use_lora, use_controlnet, use_depth, use_hed, use_ip, lora_name, lora_path, lora_weight, negative_image, neg_prompt, true_gs, guidance, cfg):
 
1
  import torch.multiprocessing as mp
2
  import multiprocessing as mp2
3
+ from huggingface_hub import login
4
+
5
  mp.set_start_method('spawn', force=True)
6
  mp2.set_start_method('spawn', force=True)
7
 
 
11
  import os
12
 
13
  import random
14
+ login()
15
 
16
  @spaces.GPU(duration=300)
17
  def process_image(image, prompt, steps, use_lora, use_controlnet, use_depth, use_hed, use_ip, lora_name, lora_path, lora_weight, negative_image, neg_prompt, true_gs, guidance, cfg):