bfloat16 vae (#456)

* bfloat16 vae

* bfloat16 vae

* bfloat16 vae
This commit is contained in:
lllyasviel
2023-09-20 08:16:20 -07:00
committed by GitHub
parent 6597b3df64
commit cdf642437c
4 changed files with 47 additions and 5 deletions
+2 -2
View File
@@ -1,7 +1,5 @@
import threading
import numpy as np
import torch
buffer = []
outputs = []
@@ -10,6 +8,8 @@ outputs = []
def worker():
global buffer, outputs
import numpy as np
import torch
import time
import shared
import random