PyTorch
Korean
English
qwen2
krx
gyung commited on
Commit
c5f69dc
โ€ข
1 Parent(s): cc72b89

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -45,7 +45,7 @@ D. 1,000,000์›์„ ์—ฐ 5% ๋ณต๋ฆฌ๋กœ 2๋…„ ๋™์•ˆ ์˜ˆ์น˜ํ–ˆ์„ ๊ฒฝ์šฐ, ๋งŒ๊ธฐ์—
45
  """
46
 
47
  # ํ† ํฐํ™” ๋ฐ ์ž…๋ ฅ ํ…์„œ๋ฅผ GPU๋กœ ์ด๋™
48
- inputs = tokenizer(input_text, return_tensors="pt").to("cuda")
49
 
50
  # ๋ชจ๋ธ์„ ์‚ฌ์šฉํ•˜์—ฌ ํ…์ŠคํŠธ ์ƒ์„ฑ
51
  output = model.generate(**inputs, max_length=1500)
 
45
  """
46
 
47
  # ํ† ํฐํ™” ๋ฐ ์ž…๋ ฅ ํ…์„œ๋ฅผ GPU๋กœ ์ด๋™
48
+ inputs = tokenizer(input_text, return_tensors="pt")
49
 
50
  # ๋ชจ๋ธ์„ ์‚ฌ์šฉํ•˜์—ฌ ํ…์ŠคํŠธ ์ƒ์„ฑ
51
  output = model.generate(**inputs, max_length=1500)