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 |
|