ten działa
This commit is contained in:
parent
a52e27920a
commit
b5ebfcc524
4
hft.py
4
hft.py
|
|
@ -15,7 +15,7 @@ from huggingface_hub import login
|
|||
# Konfiguracja
|
||||
os.environ['TORCH_USE_CUDA_DSA'] = '1'
|
||||
os.environ["TOKENIZERS_PARALLELISM"] = "false"
|
||||
login(token="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")
|
||||
login(token="TWÓJ_TOKEN_HF")
|
||||
|
||||
class SourceMapper:
|
||||
def __init__(self):
|
||||
|
|
@ -207,7 +207,7 @@ def main():
|
|||
print("\nBrak danych do treningu!")
|
||||
return
|
||||
|
||||
dataset = Dataset.from_dict({k: [dic[k] for dic in data] for k in data[0]})
|
||||
dataset = Dataset.from_list(data)
|
||||
|
||||
def tokenize_function(examples):
|
||||
tokenized = tokenizer(
|
||||
|
|
|
|||
Loading…
Reference in New Issue