xiaowang7777 commited on
Commit
51ac4d5
·
1 Parent(s): 7e35678
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from accelerate import init_empty_weights, load_checkpoint_and_dispatch
8
 
9
  # nstruct_pipeline_3b = pipeline(model="fnlp/moss-moon-003-sft-int4", torch_dtype=torch.float, trust_remote_code=True,
10
  # device_map="auto")
11
- model_path = "fnlp/moss-moon-003-sft-int4"
12
 
13
  config = MossConfig.from_pretrained(model_path)
14
  tokenizer = MossTokenizer.from_pretrained(model_path)
 
8
 
9
  # nstruct_pipeline_3b = pipeline(model="fnlp/moss-moon-003-sft-int4", torch_dtype=torch.float, trust_remote_code=True,
10
  # device_map="auto")
11
+ model_path = "fnlp/moss-moon-003-sft-int8"
12
 
13
  config = MossConfig.from_pretrained(model_path)
14
  tokenizer = MossTokenizer.from_pretrained(model_path)