Will-uob commited on
Commit
5471374
·
1 Parent(s): 673846c

Pytorch not a lib, forgot to import pytorch, fml

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +1 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import gradio as gr
 
2
  from torch import autocast
3
  from diffusers import StableDiffusionPipeline
4
 
 
1
  import gradio as gr
2
+ import torch
3
  from torch import autocast
4
  from diffusers import StableDiffusionPipeline
5
 
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  diffusers
2
  transformers
3
- pytorch
4
  numpy
5
  scipy
 
1
  diffusers
2
  transformers
3
+ torch
4
  numpy
5
  scipy