Update inference/style_transfer.py
Browse files
inference/style_transfer.py
CHANGED
@@ -325,7 +325,7 @@ class Mixing_Style_Transfer_Inference:
|
|
325 |
|
326 |
|
327 |
|
328 |
-
def set_up()
|
329 |
os.environ['MASTER_ADDR'] = '127.0.0.1'
|
330 |
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
|
331 |
os.environ['MASTER_PORT'] = '8888'
|
|
|
325 |
|
326 |
|
327 |
|
328 |
+
def set_up():
|
329 |
os.environ['MASTER_ADDR'] = '127.0.0.1'
|
330 |
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
|
331 |
os.environ['MASTER_PORT'] = '8888'
|