Spaces:
Paused
Paused
Commit
·
7843ac8
1
Parent(s):
5c1d871
Update app_v3.py
Browse files
app_v3.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
-
from transformers import AutoTokenizer, TextStreamer, pipeline
|
2 |
from auto_gptq import AutoGPTQForCausalLM, BaseQuantizeConfig
|
3 |
import time
|
4 |
|
5 |
-
model_name_or_path = "
|
6 |
-
model_basename = "gptq_model-4bit-128g"
|
7 |
|
8 |
use_triton = False
|
9 |
|
@@ -17,17 +17,6 @@ model = AutoGPTQForCausalLM.from_quantized(model_name_or_path,
|
|
17 |
use_triton=use_triton,
|
18 |
quantize_config=None)
|
19 |
|
20 |
-
"""
|
21 |
-
To download from a specific branch, use the revision parameter, as in this example:
|
22 |
-
|
23 |
-
model = AutoGPTQForCausalLM.from_quantized(model_name_or_path,
|
24 |
-
revision="gptq-4bit-32g-actorder_True",
|
25 |
-
model_basename=model_basename,
|
26 |
-
use_safetensors=True,
|
27 |
-
trust_remote_code=True,
|
28 |
-
device="cuda:0",
|
29 |
-
quantize_config=None)
|
30 |
-
"""
|
31 |
|
32 |
prompt = "Tell me about AI"
|
33 |
prompt_template=f'''### HUMAN:
|
|
|
1 |
+
from transformers import AutoTokenizer, TextStreamer, pipeline
|
2 |
from auto_gptq import AutoGPTQForCausalLM, BaseQuantizeConfig
|
3 |
import time
|
4 |
|
5 |
+
model_name_or_path = "FPHam/Jackson_The_Formalizer_V2_13b_GPTQ"
|
6 |
+
model_basename = "Jackson2-gptq_model-4bit-128g"
|
7 |
|
8 |
use_triton = False
|
9 |
|
|
|
17 |
use_triton=use_triton,
|
18 |
quantize_config=None)
|
19 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
20 |
|
21 |
prompt = "Tell me about AI"
|
22 |
prompt_template=f'''### HUMAN:
|