tuan2308 commited on
Commit
4d60c66
·
verified ·
1 Parent(s): ef623d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -8,6 +8,8 @@ BFS_LORA = "Alissonerdx/BFS-Best-Face-Swap"
8
  BFS_LORA_WEIGHT = "bfs_head_v3_qwen_image_edit_2509.safetensors" # Head V3 (recommended)
9
 
10
  # --------- PIPELINE (CPU, ZERO GPU) ---------
 
 
11
  def load_pipeline():
12
  device = "cpu"
13
  dtype = torch.float32
 
8
  BFS_LORA_WEIGHT = "bfs_head_v3_qwen_image_edit_2509.safetensors" # Head V3 (recommended)
9
 
10
  # --------- PIPELINE (CPU, ZERO GPU) ---------
11
+
12
+ @spaces.GPU
13
  def load_pipeline():
14
  device = "cpu"
15
  dtype = torch.float32