Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -43,7 +43,7 @@ class SuperDiffPipeline(DiffusionPipeline, ConfigMixin):
|
|
| 43 |
#self.scheduler = scheduler
|
| 44 |
|
| 45 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 46 |
-
|
| 47 |
vae.to(device)
|
| 48 |
unet.to(device)
|
| 49 |
text_encoder.to(device)
|
|
|
|
| 43 |
#self.scheduler = scheduler
|
| 44 |
|
| 45 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 46 |
+
print("hello this is a test", flush=True)
|
| 47 |
vae.to(device)
|
| 48 |
unet.to(device)
|
| 49 |
text_encoder.to(device)
|