Get rid of dumb print

This commit is contained in:
Galunid 2023-10-31 01:54:24 +01:00
parent 08918b700e
commit 3bb9844de9

View file

@ -95,7 +95,6 @@ class Model:
with ctx as model_part:
for name in model_part.keys():
print("yield ", name)
data = model_part.get_tensor(name) if self.is_safetensors else model_part[name]
yield name, data