Spaces:
Paused
Paused
Abdulrahman1989
commited on
Commit
·
e5eeba4
1
Parent(s):
72c6f63
print CUDA device
Browse files- SDXLImageGenerator.py +2 -0
SDXLImageGenerator.py
CHANGED
@@ -9,6 +9,8 @@ class SDXLImageGenerator:
|
|
9 |
# Set proper device based on cuda availability
|
10 |
self.device = torch.device("cuda" if self.use_cuda else "cpu")
|
11 |
|
|
|
|
|
12 |
# Load the pipeline
|
13 |
self.pipe = DiffusionPipeline.from_pretrained(
|
14 |
"stabilityai/stable-diffusion-xl-base-1.0",
|
|
|
9 |
# Set proper device based on cuda availability
|
10 |
self.device = torch.device("cuda" if self.use_cuda else "cpu")
|
11 |
|
12 |
+
print("CUDA: ", self.device)
|
13 |
+
|
14 |
# Load the pipeline
|
15 |
self.pipe = DiffusionPipeline.from_pretrained(
|
16 |
"stabilityai/stable-diffusion-xl-base-1.0",
|