Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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)
|