rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Remove debug

Browse Source
This commit is contained in:
somebody
2023-06-21 14:15:38 -05:00
parent 5f224e1366
commit aca2b532d7
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modeling/inference_models/hf_torch.py
View File

@@ -308,7 +308,6 @@ class HFTorchInferenceModel(HFInferenceModel):
with lazy_loader.use_lazy_load(dematerialized_modules=True):
metamodel = AutoModelForCausalLM.from_config(self.model_config)
tf_kwargs["device_map"] = self.breakmodel_config.get_device_map(metamodel)
print("Rodger rodger", tf_kwargs)
with lazy_loader.use_lazy_load(
enable=self.lazy_load,