Accomodate Vlad's SD.Next modules refactor (#79)
* Accomodate SD.Next modules refactor * remove secondary exception check (redundant) --------- Co-authored-by: Alex "mcmonkey" Goodwin <git_commits@alexgoodwin.dev>
This commit is contained in:
co-authored by
Alex mcmonkey Goodwin
parent
3889ebc8e3
commit
4ba4b1c03b
+9
-1
@@ -3,7 +3,15 @@ import torch
|
||||
import math
|
||||
import traceback
|
||||
from modules import shared
|
||||
from modules.models.diffusion import uni_pc
|
||||
try:
|
||||
from modules.models.diffusion import uni_pc
|
||||
except Exception as e:
|
||||
from modules.unipc import sampler
|
||||
class VladUniPC(object):
|
||||
def __init__(self):
|
||||
self.sampler = sampler
|
||||
self.uni_pc = sampler
|
||||
uni_pc = VladUniPC()
|
||||
|
||||
######################### UniPC Implementation logic #########################
|
||||
|
||||
|
||||
Reference in New Issue
Block a user