silence printing the model because it's just useless noise

This commit is contained in:
mrq 2023-03-04 16:38:24 +00:00
parent 71cc43e65c
commit 6eb7ebf847
3 changed files with 4 additions and 2 deletions

View File

@ -19,6 +19,7 @@ from datetime import datetime
from utils.util import opt_get, map_cuda_to_correct_device
import tortoise.utils.torch_intermediary as ml
def init_dist(backend, **kwargs):
# These packages have globals that screw with Windows, so only import them if needed.

View File

@ -185,7 +185,7 @@ class ExtensibleTrainer(BaseModel):
self.env['discriminators'] = self.netsD
self.env['emas'] = self.emas
self.print_network() # print network
# self.print_network() # print network
self.load() # load networks from save states as needed
# Load experiments

View File

@ -48,4 +48,5 @@ x-clip
x_transformers==1.0.4
# bitsandbytes
bitsandbytes==0.35.0
bitsandbytes==0.35.0 ; platform_system == "Windows"
bitsandbytes ; platform_system != "Windows"