Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
rachelspark committed Jan 24, 2024
1 parent 4ba4ebb commit a3f0720
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions 06_gpu_and_ml/embeddings/wikipedia/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
import json
import subprocess

from modal import Image, Secret, Stub, Volume, gpu, method, build, enter, exit
from modal import Image, Secret, Stub, Volume, build, enter, exit, gpu, method

# We first set out configuration variables for our script.
## Embedding Containers Configuration
Expand Down Expand Up @@ -127,7 +127,7 @@ class TextEmbeddingsInference:
@build()
def download_model(self):
spawn_server()

@enter()
def open_connection(self):
# If the process is running for a long time, the client does not seem to close the connections, results in a pool timeout
Expand Down

0 comments on commit a3f0720

Please sign in to comment.