You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
10 lines
289 B
Python
10 lines
289 B
Python
2 years ago
|
import torch
|
||
|
|
import os
|
||
|
|
|
||
|
|
lowvram = True if os.environ.get('LOWVRAM') == "1" else False
|
||
|
|
medvram = True if os.environ.get('MEDVRAM') == "1" else False
|
||
|
|
dtype = torch.float32 if os.environ.get('DTYPE', 'float32') else torch.float16
|
||
|
|
|
||
|
|
|
||
|
|
print("using dtype: " + os.environ.get('DTYPE', 'float32'))
|