Skip to content

Commit

Permalink
docs: fix name
Browse files Browse the repository at this point in the history
  • Loading branch information
shan-mx authored Jul 26, 2022
1 parent 3001104 commit 3603378
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions docs/user-guides/server.md
Original file line number Diff line number Diff line change
Expand Up @@ -108,10 +108,10 @@ The first thing is defining your CLIP model class which has 'init', 'encode_text

Here is the template code of xxx_model.py.
```python
from clip_server.model.clip_model import BaseCLIPModel
from clip_server.model.clip_model import CLIPModel
import torch

class YourCustomCLIPModel(BaseCLIPModel):
class YourCustomCLIPModel(CLIPModel):
def __init__(self, name: str, device: str = 'cpu', jit: bool = False, **kwargs):
super().__init__(name, **kwargs)
...
Expand All @@ -137,7 +137,7 @@ _YOUR_XXX_MODELS = {
}
```
#### 3. Import _YOUR_XXX_MODELS in clip_model.py
Look at the first line of the file:
Look at the first line of the code:
```python
from clip_server.model.pretrained_models import (
_VISUAL_MODEL_IMAGE_SIZE,
Expand Down

0 comments on commit 3603378

Please sign in to comment.