-
Hi, I would like to test your model.
but I can't find pre-trained UNet model.
It seems that Unet weights is used in your code in Pytorch-Style2paints/test.py (maybe 25_Unet_checkpoint.pth.tar). Coul…
-
I put up a quick colab hoping for this to work, but when running the demo I get an error:
Here is the [Colab](https://colab.research.google.com/github/steinhaug/stable-diffusion/blob/main/avatar/Dr…
-
Hi, I have a custom UNET in ONNX and I see that when I try to convert the model into pytorch, the method mentioned above is not implemented. Would it be difficult to implement? or is there an alternat…
-
Hi, firstly thanks for opening your source code. It is really an amazing job!
Recently, I'm trying to reproduce your code in Pytorch. And I have a question about the content revision stage, specifica…
-
## Fix the model test for `pytorch_unet.py`
1. setup env according to [Run a model under torch_xla2](https://github.com/pytorch/xla/blob/master/experimental/torch_xla2/docs/support_a_new_model.md)
…
-
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…
-
https://github.com/bubbliiiing/pspnet-pytorch/blob/f4aba6143a42b3f8c6dfdbe7a0858410227b1a29/nets/resnet.py#L64
这里的卷积核大小是3,但是unet那个repository里的resnet50卷积核大小是7?
https://github.com/bubbliiiing/unet-pyt…
-
Hi, I just installed your package via `Type 2: Copy segmentation_models_pytorch_3d folder from this repository in your project folder. `.
I discovered that certain files, particularly those in the …
-
Hello, I'm very interested in your work. I train boxnet successfully with python 3.8 , torch 2.0.1 and cuda11.7. And then I want to finetune the unet, so I set '--train_unet' True and train on the sam…
-
Hi @chao-ji ,
I was trying to convert the model from [pretrained txt2img model](https://ommer-lab.com/files/latent-diffusion/text2img.zip), but it seems that some shapes are different. I was able t…