simplify #79 further
This commit is contained in:
+1
-6
@@ -6,12 +6,7 @@ from modules import shared
|
|||||||
try:
|
try:
|
||||||
from modules.models.diffusion import uni_pc
|
from modules.models.diffusion import uni_pc
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
from modules.unipc import sampler
|
from modules import unipc as uni_pc
|
||||||
class VladUniPC(object):
|
|
||||||
def __init__(self):
|
|
||||||
self.sampler = sampler
|
|
||||||
self.uni_pc = sampler
|
|
||||||
uni_pc = VladUniPC()
|
|
||||||
|
|
||||||
######################### UniPC Implementation logic #########################
|
######################### UniPC Implementation logic #########################
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user