kasim90 commited on
Commit
b0fe154
Β·
verified Β·
1 Parent(s): 6a903dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -3,6 +3,10 @@ import os
3
  from datasets import load_dataset
4
  from transformers import AutoTokenizer, AutoModelForCausalLM, TrainingArguments, Trainer
5
  from huggingface_hub import login
 
 
 
 
6
 
7
  # Hugging Face Token'Δ± Secrets'ten al
8
  HF_TOKEN = os.getenv("HF_TOKEN")
 
3
  from datasets import load_dataset
4
  from transformers import AutoTokenizer, AutoModelForCausalLM, TrainingArguments, Trainer
5
  from huggingface_hub import login
6
+ import os
7
+ os.environ["CUDA_VISIBLE_DEVICES"] = "0"
8
+ os.environ["TORCH_CUDA_ARCH_LIST"] = ""
9
+
10
 
11
  # Hugging Face Token'Δ± Secrets'ten al
12
  HF_TOKEN = os.getenv("HF_TOKEN")