Hbhatt-merexgenAI

Results 3 comments of Hbhatt-merexgenAI

add this line after initialising pipe pipe = pipe.to("cuda") I resolved this

the training doesnt work on 12GB vram rtx 3060

Do you have an update in regards to the code release? :)