Update README.md
Browse files
README.md
CHANGED
@@ -25,8 +25,9 @@ import torch
|
|
25 |
# Load the model and tokenizer
|
26 |
model = AutoPeftModelForCausalLM.from_pretrained(
|
27 |
"LLM-SocialMedia/Qwen3-8B-Korean-Sentiment",
|
|
|
28 |
device_map="auto",
|
29 |
-
offload_folder="/offload",
|
30 |
offload_state_dict=True,
|
31 |
torch_dtype=torch.bfloat16
|
32 |
)
|
|
|
25 |
# Load the model and tokenizer
|
26 |
model = AutoPeftModelForCausalLM.from_pretrained(
|
27 |
"LLM-SocialMedia/Qwen3-8B-Korean-Sentiment",
|
28 |
+
# if GPU memory is insufficient
|
29 |
device_map="auto",
|
30 |
+
offload_folder="/offload",
|
31 |
offload_state_dict=True,
|
32 |
torch_dtype=torch.bfloat16
|
33 |
)
|