abrakjamson commited on
Commit
7197de3
1 Parent(s): fb1c1ed

link to CPU space

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ from huggingface_hub import login
16
  mistral_path = "mistralai/Mistral-7B-Instruct-v0.3"
17
 
18
  access_token = os.getenv("mistralaccesstoken")
19
- login(access_token)
20
 
21
  tokenizer = AutoTokenizer.from_pretrained(mistral_path)
22
  tokenizer.pad_token_id = 0
@@ -482,7 +482,7 @@ with gr.Blocks(
482
  else:
483
  gr.Markdown("""# 🧠 LLM Mind Control
484
 
485
- *Warning: running on CPU will be very slow*""")
486
  gr.Markdown("""Unlike prompting, direct weight manipulation lets you fine-tune the amount of a personality
487
  trait or topic. Enabled through [Representation Engineering](https://arxiv.org/abs/2310.01405)
488
  via the [repeng](https://pypi.org/project/repeng) library.
 
16
  mistral_path = "mistralai/Mistral-7B-Instruct-v0.3"
17
 
18
  access_token = os.getenv("mistralaccesstoken")
19
+ login(access_token)
20
 
21
  tokenizer = AutoTokenizer.from_pretrained(mistral_path)
22
  tokenizer.pad_token_id = 0
 
482
  else:
483
  gr.Markdown("""# 🧠 LLM Mind Control
484
 
485
+ *Warning: this space won't work well on CPU. Use the [Llama 1B version](https://huggingface.co/spaces/Abrak/Controlled_Chat_CPU) instead, or duplicate this space onto GPU hardware.""")
486
  gr.Markdown("""Unlike prompting, direct weight manipulation lets you fine-tune the amount of a personality
487
  trait or topic. Enabled through [Representation Engineering](https://arxiv.org/abs/2310.01405)
488
  via the [repeng](https://pypi.org/project/repeng) library.