Spaces:
Running
on
Zero
Running
on
Zero
tight-inversion
commited on
Commit
·
addfbd1
1
Parent(s):
d672aed
Use fp8 model
Browse files
app.py
CHANGED
@@ -453,6 +453,7 @@ if __name__ == "__main__":
|
|
453 |
parser.add_argument("--pretrained_model", type=str, help='for development')
|
454 |
args = parser.parse_args()
|
455 |
|
|
|
456 |
if args.aggressive_offload:
|
457 |
args.offload = True
|
458 |
|
|
|
453 |
parser.add_argument("--pretrained_model", type=str, help='for development')
|
454 |
args = parser.parse_args()
|
455 |
|
456 |
+
args.fp8 = True
|
457 |
if args.aggressive_offload:
|
458 |
args.offload = True
|
459 |
|