Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -56,9 +56,6 @@ pipe.tokenizer_max_length = 248
|
|
56 |
pipe.text_encoder.dtype = torch.bfloat16
|
57 |
#pipe.text_encoder_2 = t5.text_model
|
58 |
|
59 |
-
pipe.transformer.enable_layerwise_casting(
|
60 |
-
storage_dtype=torch.float8_e4m3fn, compute_dtype=torch.bfloat16
|
61 |
-
)
|
62 |
MAX_SEED = 2**32-1
|
63 |
|
64 |
class calculateDuration:
|
|
|
56 |
pipe.text_encoder.dtype = torch.bfloat16
|
57 |
#pipe.text_encoder_2 = t5.text_model
|
58 |
|
|
|
|
|
|
|
59 |
MAX_SEED = 2**32-1
|
60 |
|
61 |
class calculateDuration:
|