Update README.md

README.md

@@ -10,14 +10,14 @@ datasets:
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-device = "cuda:1" if torch.cuda.is_available() else "cpu"
+device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
 model_id = "maywell/Synatra-7B-v0.3-Translation"
 tokenizer = AutoTokenizer.from_pretrained(model_id, revision=model_revision)
 model = AutoModelForCausalLM.from_pretrained(model_id, revision=model_revision, device_map=device, torch_dtype=torch.float16).eval()
 
 # LoRA 어댑터 불러오기
-model.load_adapter("heegyu/Synatra-7B-v0.3-Translation-glaive", revision="epoch-3")
+model.load_adapter("heegyu/Synatra-7B-v0.3-Translation-glaive")
 
 
 def generate(prompt, *messages):