Skip to content

Commit 5b0e132

Browse files
committed
fix
1 parent 544098c commit 5b0e132

3 files changed

Lines changed: 4 additions & 381 deletions

File tree

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -219,8 +219,8 @@ dataset.encode(batched=True, load_from_cache_file=False)
219219
dataloader = DataLoader(dataset=dataset, batch_size=8)
220220

221221
# Initialize tinker client
222-
service_client = init_tinker_compat_client(base_model, api_key)
223-
training_client = service_client.create_lora_training_client(base_model=base_model, rank=16)
222+
service_client = init_tinker_compat_client(base_url, api_key)
223+
training_client = service_client.create_lora_training_client(base_model=base_model[len('ms://'):], rank=16)
224224

225225
# Training loop: use input_feature_to_datum to transfer the input format
226226
for epoch in range(3):

README_ZH.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -204,8 +204,8 @@ dataset.encode(batched=True, load_from_cache_file=False)
204204
dataloader = DataLoader(dataset=dataset, batch_size=8)
205205

206206
# Initialize tinker client
207-
service_client = init_tinker_compat_client(base_model, api_key)
208-
training_client = service_client.create_lora_training_client(base_model=base_model, rank=16)
207+
service_client = init_tinker_compat_client(base_url, api_key)
208+
training_client = service_client.create_lora_training_client(base_model=base_model[len('ms://'):], rank=16)
209209

210210
# Training loop: use input_feature_to_datum to transfer the input format
211211
for epoch in range(3):

tests/sequence_parallel/test_sequence_parallel_single_attention.py

Lines changed: 0 additions & 377 deletions
This file was deleted.

0 commit comments

Comments
 (0)