Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
scutcyr authored Jun 6, 2023
1 parent 9a7b406 commit 7a5e7a8
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,21 @@ conda activate proactivehealthgpt_py38
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
```

* 启动服务
本项目提供了[soulchat_app.py](./soulchat_app.py)作为SoulChat模型的使用示例,通过以下命令即可开启服务,然后,通过http://<your_ip>:9026访问。
```bash
streamlit run soulchat_app.py --server.port 9026
```
特别地,在[soulchat_app.py](./soulchat_app.py)当中,
可以修改以下代码更换指定的显卡:
```python
os.environ['CUDA_VISIBLE_DEVICES'] = '2'
```
可以通过更改以下代码指定模型路径为本地路径:
```python
model_name_or_path = 'scutcyr/SoulChat'
```


## 示例
* 样例1:失恋
Expand Down

0 comments on commit 7a5e7a8

Please sign in to comment.