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:
Fred Ostrander
2023-10-18 18:04:21 -07:00
committed by GitHub
co-authored by Alex mcmonkey Goodwin
parent 3889ebc8e3
commit 4ba4b1c03b
+9 -1
View File
@@ -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 #########################