Quantumbraid's picture
download
raw
704 Bytes
import os
from collections import OrderedDict
from jobs import BaseJob
from toolkit.metadata import get_meta_for_safetensors
from toolkit.train_tools import get_torch_dtype
process_dict = {
'rescale_lora': 'ModRescaleLoraProcess',
}
class ModJob(BaseJob):
def __init__(self, config: OrderedDict):
super().__init__(config)
self.device = self.get_conf('device', 'cpu')
# loads the processes from the config
self.load_processes(process_dict)
def run(self):
super().run()
print("")
print(f"Running {len(self.process)} process{'' if len(self.process) == 1 else 'es'}")
for process in self.process:
process.run()

Xet Storage Details

Size:
704 Bytes
·
Xet hash:
f9592546d93aa037ed488e608f387c0ad82c639bd80f031501847a30b1b0a026

Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.