Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ from transformers import (
|
|
11 |
StoppingCriteriaList
|
12 |
)
|
13 |
|
14 |
-
MODEL_ID ="Qwen/Qwen2.5-Coder-32B-Instruct"
|
15 |
|
16 |
DEFAULT_SYSTEM_PROMPT = """
|
17 |
You are an Expert AI Coding Assistant, capable of solving complex problems and crafting efficient, reliable solutions. Follow this structured workflow for every task:
|
@@ -69,7 +69,7 @@ def initialize_model():
|
|
69 |
model = AutoModelForCausalLM.from_pretrained(
|
70 |
MODEL_ID,
|
71 |
device_map="cuda",
|
72 |
-
quantization_config=quantization_config,
|
73 |
torch_dtype=torch.bfloat16,
|
74 |
trust_remote_code=True
|
75 |
).to("cuda")
|
|
|
11 |
StoppingCriteriaList
|
12 |
)
|
13 |
|
14 |
+
MODEL_ID ="Qwen/Qwen2.5-Coder-32B-Instruct-AWQ"
|
15 |
|
16 |
DEFAULT_SYSTEM_PROMPT = """
|
17 |
You are an Expert AI Coding Assistant, capable of solving complex problems and crafting efficient, reliable solutions. Follow this structured workflow for every task:
|
|
|
69 |
model = AutoModelForCausalLM.from_pretrained(
|
70 |
MODEL_ID,
|
71 |
device_map="cuda",
|
72 |
+
#quantization_config=quantization_config,
|
73 |
torch_dtype=torch.bfloat16,
|
74 |
trust_remote_code=True
|
75 |
).to("cuda")
|