NTIS's picture
Upload checkpoint-16000
8a562a2 verified
metadata
license: apache-2.0
language:
  - ko
  - en
tags:
  - text-generation
  - pytorch
  - causal-lm
library_name: transformers

gemma3-1b-cpt-final-checkpoint-16000

์ด ๋ชจ๋ธ์€ ํŒŒ์ธํŠœ๋‹๋œ ์–ธ์–ด ๋ชจ๋ธ ์ฒดํฌํฌ์ธํŠธ์ž…๋‹ˆ๋‹ค.

๋ชจ๋ธ ์ •๋ณด

  • ๋ฒ ์ด์Šค ๋ชจ๋ธ: gemma3-1b-cpt-final
  • ์ฒดํฌํฌ์ธํŠธ: checkpoint-16000
  • ํƒ€์ž…: Causal Language Model
  • ๋ผ์ด์„ ์Šค: Apache 2.0

์‚ฌ์šฉ ๋ฐฉ๋ฒ•

from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

model_name = "NTIS/gemma3-1b-cpt-final-checkpoint-16000"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(
    model_name,
    torch_dtype=torch.float16,
    device_map="auto"
)

# ํ…์ŠคํŠธ ์ƒ์„ฑ
text = "์•ˆ๋…•ํ•˜์„ธ์š”"
inputs = tokenizer(text, return_tensors="pt")
outputs = model.generate(**inputs, max_length=100, do_sample=True, temperature=0.7)
result = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(result)

์ฃผ์˜์‚ฌํ•ญ

  • ์ด ๋ชจ๋ธ์€ ์—ฐ๊ตฌ/์‹คํ—˜ ๋ชฉ์ ์œผ๋กœ ์ œ๊ณต๋ฉ๋‹ˆ๋‹ค
  • ์ƒ์—…์  ์‚ฌ์šฉ ์ „์— ๋ผ์ด์„ ์Šค๋ฅผ ํ™•์ธํ•˜์„ธ์š”