vltnmmdv commited on
Commit
f9421a4
·
verified ·
1 Parent(s): c6dcbda

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -24,7 +24,7 @@ pip install --upgrade transformers torch accelerate bitsandbytes
24
  import torch
25
  from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
26
 
27
- model_name = "vltnmmdv/GigaChat-20B-A3B-instruct-int8"
28
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
29
  model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True, device_map="auto")
30
  model.generation_config = GenerationConfig.from_pretrained(model_name)
 
24
  import torch
25
  from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
26
 
27
+ model_name = "ai-sage/GigaChat-20B-A3B-instruct-int8"
28
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
29
  model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True, device_map="auto")
30
  model.generation_config = GenerationConfig.from_pretrained(model_name)