fffiloni commited on
Commit
e7db0ca
·
verified ·
1 Parent(s): 8666f4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ try:
42
  raise FileNotFoundError(f"LoRA model not found at: {LORA_MODEL_PATH}")
43
 
44
  print(f"Is CUDA available: {torch.cuda.is_available()}")
45
- if torch.cuda.is_available():
46
  face_processor = FaceProcessor(antelopv2_path=ANTELOPEV2_PATH)
47
  pipe = load_wan_pipe(base_path=BASE_MODEL_PATH, torch_dtype=torch.bfloat16)
48
  set_stand_in(pipe, model_path=LORA_MODEL_PATH)
 
42
  raise FileNotFoundError(f"LoRA model not found at: {LORA_MODEL_PATH}")
43
 
44
  print(f"Is CUDA available: {torch.cuda.is_available()}")
45
+ if torch.cuda.is_available() and not is_shared_ui:
46
  face_processor = FaceProcessor(antelopv2_path=ANTELOPEV2_PATH)
47
  pipe = load_wan_pipe(base_path=BASE_MODEL_PATH, torch_dtype=torch.bfloat16)
48
  set_stand_in(pipe, model_path=LORA_MODEL_PATH)