-
Hello, I use my own data set under the 16g memory of the graphics card. If I change the batchsize to 2, it will cause cuda oom in LightM-Unet. How to deal with it?
2024-04-25 09:17:44.026392: unpacki…
-
## ❓ Question
How do you save a unet model compiled Torch-TensorRT from Stable Diffusion XL?
## What you have already tried
I've tried following the compilation instructions from the tutorial…
dru10 updated
2 months ago
-
Code:
```python
torch._dynamo.config.suppress_errors = True
self.pipe = OotdPipeline.from_pretrained(
MODEL_PATH,
unet_garm=unet_garm,
unet_vton=unet_vton,
vae=vae,
torch…
-
raceback (most recent call last):
File "/data/Unique3D/app/gradio_local.py", line 20, in
from app.gradio_3dgen import create_ui as create_3d_ui
File "/data/Unique3D/./app/gradio_3dgen.py",…
-
Was working great til todays update.
# ComfyUI Error Report
## Error Details
- **Node Type:** FlowStateUnifiedModelLoader
- **Exception Type:** TypeError
- **Exception Message:** object of …
-
pretrained checkpoint :
output_dir: "outputs_ctrl_flow_gen"
pretrained_model_path: "models/stage1/StableDiffusion-FlowGen/"
vae_pretrained_path: "models/stage1/StableDiffusion-FlowGen/vae_flow/di…
-
### Expected Behavior
KSampler
Given groups=1, weight of size [320, 4, 3, 3], expected input[2, 16, 90, 62] to have 4 channels, but got 16 channels instead
### Actual Behavior
KSampler
Given grou…
-
Hello, pytorch-unet has been updated to version 3.0 and supports multi-classification. Can you update the code of tensorrt-unet? thank you very much
-
# 3D UNet Mouse Embryo Live
BioImage.IO -- an AI model repository for deep learning.
[https://bioimage.io/?tags=pytorch&id=10.5281%2Fzenodo.6384845&type=model](https://bioimage.io/?tags=pytorch&id=1…
-
Just keeps saying "Enabling PyTorch fallback as no engine was found" when I try and use a TensorRT engine for a SDXL model. Tried reinstalling the extension, deleting venv and starting over and rebuil…