Hugging Face's logo

Spaces:
TencentARC
/
PhotoMaker-V2
Running on Zero

App Files Community
34

cuda_init_error

#34
by raphael-gl HF Staff - opened
base: refs/heads/main
←
from: refs/pr/34
Discussion Files changed
+5
-4
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +4 -3
app.py CHANGED
@@ -21,7 +21,7 @@ from aspect_ratio_template import aspect_ratios
21
 
22
  # global variable
23
  base_model_path = 'SG161222/RealVisXL_V4.0'
24
- face_detector = FaceAnalysis2(providers=['CPUExecutionProvider', 'CUDAExecutionProvider'], allowed_modules=['detection', 'recognition'])
25
  face_detector.prepare(ctx_id=0, det_size=(640, 640))
26
 
27
  try:
 
21
 
22
  # global variable
23
  base_model_path = 'SG161222/RealVisXL_V4.0'
24
+ face_detector = FaceAnalysis2(providers=['CPUExecutionProvider'], allowed_modules=['detection', 'recognition'])
25
  face_detector.prepare(ctx_id=0, det_size=(640, 640))
26
 
27
  try:
requirements.txt CHANGED
@@ -1,7 +1,8 @@
1
  diffusers==0.29.0
2
- torch
 
3
  torchvision
4
- transformers
5
  accelerate
6
  safetensors
7
  einops
@@ -11,4 +12,4 @@ omegaconf
11
  peft
12
  huggingface-hub
13
  insightface
14
- numpy
 
1
  diffusers==0.29.0
2
+ # Blackwell support
3
+ torch>=2.8.0
4
  torchvision
5
+ transformers<4.52
6
  accelerate
7
  safetensors
8
  einops
 
12
  peft
13
  huggingface-hub
14
  insightface
15
+ numpy