rydlrKE commited on
Commit
8cc7ff1
·
1 Parent(s): 782371a

Spaces: import spaces early in text-encoder server to avoid CUDA init warning thread crash

Browse files
kimodo/scripts/run_text_encoder_server.py CHANGED
@@ -4,6 +4,12 @@
4
  import argparse
5
  import os
6
 
 
 
 
 
 
 
7
  import gradio as gr
8
  import numpy as np
9
 
 
4
  import argparse
5
  import os
6
 
7
+ # Import spaces as early as possible in Spaces runtimes.
8
+ try:
9
+ import spaces # noqa: F401
10
+ except Exception:
11
+ pass
12
+
13
  import gradio as gr
14
  import numpy as np
15