Spaces:

MOSS550V
/

divination

Paused

MOSS550V commited on Jun 9, 2023

Commit

8a6bba4

1 Parent(s): 7d6fb66

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -125,7 +125,7 @@ def main():
     tokenizer = AutoTokenizer.from_pretrained(
         "THUDM/chatglm-6b-int4", trust_remote_code=True)
     config = AutoConfig.from_pretrained(
-        "MOSS550V/divination", trust_remote_code=True)
     config.pre_seq_len = 128
     config.prefix_projection = False

     tokenizer = AutoTokenizer.from_pretrained(
         "THUDM/chatglm-6b-int4", trust_remote_code=True)
     config = AutoConfig.from_pretrained(
+        "checkpoint-500", trust_remote_code=True)
     config.pre_seq_len = 128
     config.prefix_projection = False