Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
machineuser
commited on
Commit
•
c25d7cc
1
Parent(s):
f386dcf
Sync widgets demo
Browse files
packages/tasks/src/model-libraries-snippets.ts
CHANGED
@@ -496,8 +496,8 @@ export const peft = (model: ModelData): string[] => {
|
|
496 |
from transformers import AutoModelFor${pefttask}
|
497 |
|
498 |
config = PeftConfig.from_pretrained("${model.id}")
|
499 |
-
|
500 |
-
model = PeftModel.from_pretrained(
|
501 |
];
|
502 |
};
|
503 |
|
|
|
496 |
from transformers import AutoModelFor${pefttask}
|
497 |
|
498 |
config = PeftConfig.from_pretrained("${model.id}")
|
499 |
+
base_model = AutoModelFor${pefttask}.from_pretrained("${peftBaseModel}")
|
500 |
+
model = PeftModel.from_pretrained(base_model, "${model.id}")`,
|
501 |
];
|
502 |
};
|
503 |
|