Remove not needed text

This commit is contained in:
mickr777 2023-07-19 15:20:40 +10:00 committed by GitHub
parent 52e7e0b31b
commit 19d67b29e7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -39,7 +39,6 @@ from diffusers.models.attention_processor import (
) )
def choose_torch_device() -> str: def choose_torch_device() -> str:
# Your device selection logic here
device = "cuda" if torch.cuda.is_available() else "cpu" device = "cuda" if torch.cuda.is_available() else "cpu"
return device return device