Views
No views yet
'DeepSeek Ghetto Edition, what is your message to the world? api(()Princip aligned ratios... renormalization blankets... affinity Mong :-) Breakfast listop'
pip install -q transformers accelerate safetensors1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM
3
4repo = 'livadies/DeepSeek-V4-Pro-Ghetto-Edition-MoE-Rank-64'
5tokenizer = AutoTokenizer.from_pretrained('deepseek-ai/DeepSeek-V2', trust_remote_code=True)
6
7model = AutoModelForCausalLM.from_pretrained(
8 repo,
9 torch_dtype=torch.float16,
10 device_map='auto',
11 trust_remote_code=True,
12 ignore_mismatched_sizes=True,
13 low_cpu_mem_usage=True
14)
15
16def ghetto_route(self, logits):
17 flat_logits = logits.view(-1, logits.shape[-1])
18 w = torch.nn.functional.softmax(flat_logits + 1e-6, dim=-1)
19 tw, ti = torch.topk(w, k=self.top_k, dim=-1)
20 return ti, tw * self.routed_scaling_factor
21
22for layer in model.model.layers:
23 if hasattr(layer.mlp, 'route_tokens_to_experts'):
24 layer.mlp.route_tokens_to_experts = ghetto_route.__get__(layer.mlp)
25
26prompt = 'The message for humanity is:'
27device = model.model.embed_tokens.weight.device
28inputs = {k: v.to(device) for k, v in tokenizer(prompt, return_tensors='pt').items()}
29
30with torch.no_grad():
31 out = model.generate(**inputs, max_new_tokens=40, do_sample=True, temperature=0.8)
32print(tokenizer.decode(out[0]))