Spaces:
Runtime error
Runtime error
Raman Dutt
commited on
Commit
·
a9e7413
1
Parent(s):
0aa83a7
minor
Browse files
app.py
CHANGED
|
@@ -16,6 +16,8 @@ OUTPUT_DIR = "OUTPUT"
|
|
| 16 |
cuda_device = 1
|
| 17 |
device = f"cuda:{cuda_device}" if torch.cuda.is_available() else "cpu"
|
| 18 |
|
|
|
|
|
|
|
| 19 |
TITLE = "Demo for Generating Chest X-rays using Diferent Parameter-Efficient Fine-Tuned Stable Diffusion Pipelines"
|
| 20 |
INFO_ABOUT_TEXT_PROMPT = "INFO_ABOUT_TEXT_PROMPT"
|
| 21 |
INFO_ABOUT_GUIDANCE_SCALE = "INFO_ABOUT_GUIDANCE_SCALE"
|
|
|
|
| 16 |
cuda_device = 1
|
| 17 |
device = f"cuda:{cuda_device}" if torch.cuda.is_available() else "cpu"
|
| 18 |
|
| 19 |
+
print("DEVICE: ", device)
|
| 20 |
+
|
| 21 |
TITLE = "Demo for Generating Chest X-rays using Diferent Parameter-Efficient Fine-Tuned Stable Diffusion Pipelines"
|
| 22 |
INFO_ABOUT_TEXT_PROMPT = "INFO_ABOUT_TEXT_PROMPT"
|
| 23 |
INFO_ABOUT_GUIDANCE_SCALE = "INFO_ABOUT_GUIDANCE_SCALE"
|