Spaces:
Running
on
Zero
Running
on
Zero
Alexander Bagus
commited on
Commit
·
664c8fa
1
Parent(s):
265bf34
22
Browse files
app.py
CHANGED
|
@@ -31,6 +31,7 @@ MAX_IMAGE_SIZE = 1280
|
|
| 31 |
DTYPE = torch.bfloat16
|
| 32 |
|
| 33 |
# load transformer
|
|
|
|
| 34 |
transformer = ZImageControlTransformer2DModel.from_pretrained(
|
| 35 |
MODEL_LOCAL,
|
| 36 |
subfolder="transformer",
|
|
|
|
| 31 |
DTYPE = torch.bfloat16
|
| 32 |
|
| 33 |
# load transformer
|
| 34 |
+
print(TRANSFORMER_CONFIG)
|
| 35 |
transformer = ZImageControlTransformer2DModel.from_pretrained(
|
| 36 |
MODEL_LOCAL,
|
| 37 |
subfolder="transformer",
|