xiaowang7777 commited on
Commit
3315b61
1 Parent(s): 0c0f97c
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
9
 
10
  # nstruct_pipeline_3b = pipeline(model="fnlp/moss-moon-003-sft-int4", torch_dtype=torch.float, trust_remote_code=True,
11
  # device_map="auto")
12
- model_path = "fnlp/moss-moon-003-sft-int8"
13
 
14
  # config = MossConfig.from_pretrained(model_path)
15
  # tokenizer = MossTokenizer.from_pretrained(model_path)
 
9
 
10
  # nstruct_pipeline_3b = pipeline(model="fnlp/moss-moon-003-sft-int4", torch_dtype=torch.float, trust_remote_code=True,
11
  # device_map="auto")
12
+ model_path = "fnlp/moss-moon-003-sft-int4"
13
 
14
  # config = MossConfig.from_pretrained(model_path)
15
  # tokenizer = MossTokenizer.from_pretrained(model_path)