fix bug of florence2 model loader

This commit is contained in:
chflame163
2024-08-23 10:51:32 +08:00
parent 31253dfb30
commit 65820e056a
2 changed files with 2 additions and 3 deletions
+1 -2
View File
@@ -30,7 +30,7 @@ from PIL import Image, ImageFilter, ImageChops, ImageDraw, ImageOps, ImageEnhanc
from skimage import img_as_float, img_as_ubyte
import torchvision.transforms.functional as TF
import torch.nn.functional as F
from transformers import StoppingCriteria, StoppingCriteriaList
from transformers import AutoModel, AutoProcessor, StoppingCriteria, StoppingCriteriaList, AutoModelForCausalLM
from colorsys import rgb_to_hsv
import folder_paths
from .filmgrainer import processing as processing_utils
@@ -1997,7 +1997,6 @@ class UformGen2QwenChat:
def __init__(self):
from huggingface_hub import snapshot_download
from transformers import AutoModel, AutoProcessor
# self.model_path = snapshot_download("unum-cloud/uform-gen2-qwen-500m",
# local_dir=files_for_uform_gen2_qwen,
# force_download=False, # Set to True if you always want to download, regardless of local copy