Daemontatox commited on
Commit
4f03c98
·
verified ·
1 Parent(s): 9c46858

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -20,7 +20,7 @@ from transformers import (
20
  )
21
 
22
  # Configuration Constants
23
- MODEL_ID = "Daemontatox/Sphinx2.0"
24
  DEFAULT_SYSTEM_PROMPT = """You are a Sentient Reasoning AI , You Provide quality and Accurate Answers to the user."""
25
 
26
  # UI Configuration
@@ -82,8 +82,8 @@ def initialize_model():
82
  torch_dtype=torch.float16,
83
  device_map="auto",
84
  attn_implementation="flash_attention_2",
85
- quantization_config=quantization_config,
86
- llm_int8_enable_fp32_cpu_offload=True
87
  )
88
 
89
  return model, tokenizer
 
20
  )
21
 
22
  # Configuration Constants
23
+ MODEL_ID = "Daemontatox/AetherSett"
24
  DEFAULT_SYSTEM_PROMPT = """You are a Sentient Reasoning AI , You Provide quality and Accurate Answers to the user."""
25
 
26
  # UI Configuration
 
82
  torch_dtype=torch.float16,
83
  device_map="auto",
84
  attn_implementation="flash_attention_2",
85
+ # quantization_config=quantization_config
86
+
87
  )
88
 
89
  return model, tokenizer