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

colab debug

Browse Source
This commit is contained in:
somebody
2023-02-27 18:40:26 -06:00
parent 2e3ca6f769
commit 1839de1483
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
model.py
View File

@@ -2203,6 +2203,8 @@ class GenericHFTorchInferenceModel(HFTorchInferenceModel):
# Get the model_type from the config or assume a model type if it isn't present
try:
print("LMP:", self.get_local_model_path())
print("M:", utils.koboldai_vars.model)
model_config = AutoConfig.from_pretrained(
self.get_local_model_path() or utils.koboldai_vars.model,
revision=utils.koboldai_vars.revision,