jhtonyKoo commited on
Commit
8c11be7
1 Parent(s): 94bab1b

Update inference/style_transfer.py

Browse files
Files changed (1) hide show
  1. inference/style_transfer.py +1 -1
inference/style_transfer.py CHANGED
@@ -301,7 +301,7 @@ class Mixing_Style_Transfer_Inference:
301
  return whole_batch_data
302
 
303
 
304
- def set_up()
305
  os.environ['MASTER_ADDR'] = '127.0.0.1'
306
  os.environ["CUDA_VISIBLE_DEVICES"] = '0'
307
  os.environ['MASTER_PORT'] = '8888'
 
301
  return whole_batch_data
302
 
303
 
304
+ def set_up():
305
  os.environ['MASTER_ADDR'] = '127.0.0.1'
306
  os.environ["CUDA_VISIBLE_DEVICES"] = '0'
307
  os.environ['MASTER_PORT'] = '8888'