Update inference_manager.py
Browse files- inference_manager.py +2 -1
inference_manager.py
CHANGED
|
@@ -3,6 +3,7 @@ import os
|
|
| 3 |
import json
|
| 4 |
import time
|
| 5 |
import copy
|
|
|
|
| 6 |
import torch
|
| 7 |
import random
|
| 8 |
from diffusers import AutoPipelineForText2Image, StableDiffusionPipeline,DiffusionPipeline, StableDiffusionXLPipeline, AutoencoderKL, AutoencoderTiny, UNet2DConditionModel
|
|
@@ -25,7 +26,7 @@ import cv2
|
|
| 25 |
import re
|
| 26 |
import gradio as gr
|
| 27 |
from PIL import Image
|
| 28 |
-
|
| 29 |
#from onediffx import compile_pipe, save_pipe, load_pipe
|
| 30 |
|
| 31 |
HF_TOKEN = os.getenv('HF_TOKEN')
|
|
|
|
| 3 |
import json
|
| 4 |
import time
|
| 5 |
import copy
|
| 6 |
+
import numpy as np
|
| 7 |
import torch
|
| 8 |
import random
|
| 9 |
from diffusers import AutoPipelineForText2Image, StableDiffusionPipeline,DiffusionPipeline, StableDiffusionXLPipeline, AutoencoderKL, AutoencoderTiny, UNet2DConditionModel
|
|
|
|
| 26 |
import re
|
| 27 |
import gradio as gr
|
| 28 |
from PIL import Image
|
| 29 |
+
MAX_SEED = np.iinfo(np.int32).max
|
| 30 |
#from onediffx import compile_pipe, save_pipe, load_pipe
|
| 31 |
|
| 32 |
HF_TOKEN = os.getenv('HF_TOKEN')
|