Skip to content

GitLab

"src/diffusers/models/unets/unet_2d.py" did not exist on "4c8a05f1159b6bacb78b608f32fd97ffe80ea59d"
Unverified Commit c32bec56 authored by Omkar Salpekar's avatar Omkar Salpekar Committed by GitHub
Browse files

[Nova] More comprehensive Smoke Tests for Torchvision (#6803) 

* [Nova] More comprehensive Smoke Tests for Torchvision

* No need to import everything since nothing is lazily uploaded
parent e47843c5
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
test/smoke_test.py
View file @ c32bec56
import torch """Run smoke tests"""
import torchvision import torchvision
import torchvision.datasets as dset
import torchvision.transforms print("torchvision version is ", torchvision.__version__)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment