Spaces:

pythainlp
/

wangchanglm-demo-cpu

Running

wannaphong commited on May 29, 2023

Commit

511d5d9

1 Parent(s): 2ae4509

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -125,7 +125,7 @@ f.close()
 guardian = SensitiveTopicProtector.fromRaw(sensitive_topics)
-name_model = "pythainlp/wangchanglm-7.5B-sft-en-sharded"
 model = AutoModelForCausalLM.from_pretrained(
     name_model,
     device_map="auto",

 guardian = SensitiveTopicProtector.fromRaw(sensitive_topics)
+name_model = "pythainlp/wangchanglm-7.5B-sft-en-8bit-sharded"
 model = AutoModelForCausalLM.from_pretrained(
     name_model,
     device_map="auto",