Spaces:
Running
on
Zero
Running
on
Zero
Daemontatox
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ from transformers import (
|
|
20 |
)
|
21 |
|
22 |
# Configuration Constants
|
23 |
-
MODEL_ID = "Daemontatox/
|
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 |
-
|
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
|