Spaces:

orangewong
/

Zhi-writing-dsr1-14b-test

Running on Zero

orangewong commited on May 4

Commit

94a08bb

verified ·

1 Parent(s): c4b00c7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 import spaces
-model_name = "Zhihu-ai/Zhi-writing-dsr1-14"
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
     model_name,

 import spaces
+model_name = "Zhihu-ai/Zhi-writing-dsr1-14b"
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
     model_name,