TimS commited on
Commit
4e6d581
·
1 Parent(s): 7b2c744
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ args = parser.parse_args()
42
 
43
  if HF_SPACE:
44
  model = FragLMLlamaModel.from_pretrained(HF_MODEL, token=os.getenv('HF_TOKEN')).to(device)
45
- designer = FragLMDesign(model=model, tokenizer=HF_MODEL)
46
  else:
47
  model = FragLMLlamaModel.from_pretrained(args.model_path).to(device)
48
  designer = FragLMDesign(model=model, tokenizer=args.tokenizer_path)
 
42
 
43
  if HF_SPACE:
44
  model = FragLMLlamaModel.from_pretrained(HF_MODEL, token=os.getenv('HF_TOKEN')).to(device)
45
+ designer = FragLMDesign(model=model, tokenizer="/data/fraglm/tokenizer/fraglm_2406_bpe_8k.json")
46
  else:
47
  model = FragLMLlamaModel.from_pretrained(args.model_path).to(device)
48
  designer = FragLMDesign(model=model, tokenizer=args.tokenizer_path)