mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-08 22:48:25 -05:00
hotfix: save sdxl ram
This commit is contained in:
@@ -389,6 +389,7 @@ if __name__ == "__main__":
|
||||
start_mem_used = GlobalCounters.mem_used
|
||||
with Timing("loaded weights in ", lambda et_ns: f", {(B:=(GlobalCounters.mem_used-start_mem_used))/1e9:.2f} GB loaded at {B/et_ns:.2f} GB/s"):
|
||||
Tensor.realize(*loaded_weights)
|
||||
del loaded_weights
|
||||
|
||||
N = 1
|
||||
C = 4
|
||||
|
||||
Reference in New Issue
Block a user