tommytracx commited on
Commit
36a53a4
·
verified ·
1 Parent(s): 5f5869e

Add configuration_ollama.py

Browse files
Files changed (1) hide show
  1. configuration_ollama.py +74 -0
configuration_ollama.py ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ NeuralQuantum Ollama Configuration for Hugging Face Transformers
3
+ """
4
+
5
+ from transformers import PretrainedConfig
6
+
7
+
8
+ class NeuralQuantumOllamaConfig(PretrainedConfig):
9
+ """Configuration class for NeuralQuantum Ollama model"""
10
+
11
+ model_type = "neuralquantum_ollama"
12
+
13
+ def __init__(
14
+ self,
15
+ vocab_size=50257,
16
+ hidden_size=768,
17
+ num_attention_heads=12,
18
+ num_hidden_layers=12,
19
+ intermediate_size=3072,
20
+ hidden_act="gelu",
21
+ hidden_dropout_prob=0.1,
22
+ attention_probs_dropout_prob=0.1,
23
+ max_position_embeddings=2048,
24
+ type_vocab_size=2,
25
+ initializer_range=0.02,
26
+ layer_norm_eps=1e-12,
27
+ use_cache=True,
28
+ quantum_enhancement=True,
29
+ quantum_layers=6,
30
+ quantum_circuit_depth=12,
31
+ quantum_optimization="vqe",
32
+ hybrid_mode=True,
33
+ ollama_optimized=True,
34
+ temperature=0.7,
35
+ top_p=0.9,
36
+ top_k=40,
37
+ repeat_penalty=1.1,
38
+ num_ctx=2048,
39
+ num_predict=512,
40
+ torch_dtype="float16",
41
+ **kwargs
42
+ ):
43
+ super().__init__(**kwargs)
44
+
45
+ self.vocab_size = vocab_size
46
+ self.hidden_size = hidden_size
47
+ self.num_attention_heads = num_attention_heads
48
+ self.num_hidden_layers = num_hidden_layers
49
+ self.intermediate_size = intermediate_size
50
+ self.hidden_act = hidden_act
51
+ self.hidden_dropout_prob = hidden_dropout_prob
52
+ self.attention_probs_dropout_prob = attention_probs_dropout_prob
53
+ self.max_position_embeddings = max_position_embeddings
54
+ self.type_vocab_size = type_vocab_size
55
+ self.initializer_range = initializer_range
56
+ self.layer_norm_eps = layer_norm_eps
57
+ self.use_cache = use_cache
58
+
59
+ # Quantum-specific parameters
60
+ self.quantum_enhancement = quantum_enhancement
61
+ self.quantum_layers = quantum_layers
62
+ self.quantum_circuit_depth = quantum_circuit_depth
63
+ self.quantum_optimization = quantum_optimization
64
+ self.hybrid_mode = hybrid_mode
65
+ self.ollama_optimized = ollama_optimized
66
+
67
+ # Ollama-specific parameters
68
+ self.temperature = temperature
69
+ self.top_p = top_p
70
+ self.top_k = top_k
71
+ self.repeat_penalty = repeat_penalty
72
+ self.num_ctx = num_ctx
73
+ self.num_predict = num_predict
74
+ self.torch_dtype = torch_dtype