File size: 1,138 Bytes
22ca508
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
version: '3.8'

services:
  gpt-local:
    build: .
    container_name: gpt-local
    ports:
      - "7860:7860"
    volumes:
      # Persistir cache de modelos
      - ./models_cache:/app/models_cache
      - ~/.cache/huggingface:/root/.cache/huggingface
    environment:
      - PYTHONPATH=/app
      - HF_HOME=/root/.cache/huggingface
      - HUGGINGFACE_TOKEN=${HUGGINGFACE_TOKEN}
      - GITHUB_TOKEN=${GITHUB_TOKEN}
      - DEFAULT_MODEL=${DEFAULT_MODEL:-microsoft/DialoGPT-small}
      - DEVICE=${DEVICE:-auto}
    # Para chat en terminal
    tty: true
    stdin_open: true
    command: python3 chat_terminal.py

  gpt-local-web:
    build: .
    container_name: gpt-local-web
    ports:
      - "7860:7860"
    volumes:
      - ./models_cache:/app/models_cache
      - ~/.cache/huggingface:/root/.cache/huggingface
    environment:
      - PYTHONPATH=/app
      - HF_HOME=/root/.cache/huggingface
      - HUGGINGFACE_TOKEN=${HUGGINGFACE_TOKEN}
      - GITHUB_TOKEN=${GITHUB_TOKEN}
      - DEFAULT_MODEL=${DEFAULT_MODEL:-microsoft/DialoGPT-small}
      - DEVICE=${DEVICE:-auto}
    # Para interfaz web
    command: python3 main.py