import torchfrom transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig model_name = "ai-sage/GigaChat-20B-A3B-base"tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True, torch_dtype=torch.bfloat16, device_map="auto")model.generation_config = GenerationConfig.from_pretrained(model_name) messages = ( "Ниже я написал подробное доказательство теоремы о неподвижной точке:")input_tensor = tokenizer(messages, return_tensors="pt").input_idsoutputs = model.generate(input_tensor.to(model.device)) result = tokenizer.decode(outputs[0][input_tensor.shape[1]:], skip_special_tokens=False)print(result)