Hugging Face's logo

Spaces:
mpatel57
/
WOUAF-Text-to-Image
Running on T4

App Files Community
1
Maitreyapatel commited on
Commit
3078956
•
1 Parent(s): eb4d373

bug resolved

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +2 -0
app.py CHANGED
@@ -49,7 +49,7 @@ class AttributionModel:
49
  )
50
  self.vae = customize_vae_decoder(self.vae, 128, "qkv", "all", False, 1.0)
51
 
52
- self.mapping_network = MappingNetwork(32, 0, 128, None, num_layers=2, w_avg_beta=None, normalization = False).to("cuda")
53
 
54
  from torchvision.models import resnet50, ResNet50_Weights
55
  self.decoding_network = resnet50(weights=ResNet50_Weights.IMAGENET1K_V2)
 
49
  )
50
  self.vae = customize_vae_decoder(self.vae, 128, "qkv", "all", False, 1.0)
51
 
52
+ self.mapping_network = MappingNetwork(32, 0, 128, None, num_layers=2, w_avg_beta=None, normalization = False)
53
 
54
  from torchvision.models import resnet50, ResNet50_Weights
55
  self.decoding_network = resnet50(weights=ResNet50_Weights.IMAGENET1K_V2)
requirements.txt CHANGED
@@ -1,4 +1,6 @@
 
1
  torchvision
 
2
  diffusers==0.16.0
3
  transformers
4
  accelerate==0.18.0
 
1
+ torch
2
  torchvision
3
+ torchaudio
4
  diffusers==0.16.0
5
  transformers
6
  accelerate==0.18.0