Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
caaeec1
1
Parent(s):
4ebb621
Update app.py
Browse files
app.py
CHANGED
|
@@ -48,7 +48,7 @@ pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtyp
|
|
| 48 |
#pipe.num_single_layers="0"
|
| 49 |
pipe.transformer_chunk_size="0"
|
| 50 |
#model.pooled_projections="(_, 1)[0]"
|
| 51 |
-
pipe.transformer_pooled_projections_dim="0"
|
| 52 |
|
| 53 |
pipe.to("cuda")
|
| 54 |
#clipmodel = 'norm'
|
|
|
|
| 48 |
#pipe.num_single_layers="0"
|
| 49 |
pipe.transformer_chunk_size="0"
|
| 50 |
#model.pooled_projections="(_, 1)[0]"
|
| 51 |
+
pipe.transformer_pooled_projections_dim="(batch_size, 0)"
|
| 52 |
|
| 53 |
pipe.to("cuda")
|
| 54 |
#clipmodel = 'norm'
|