Hugging Face's logo

Spaces:
ruslanmv
/
Text-to-Images
Running on Zero

App Files Community
ruslanmv commited on
Commit
ff5d480
·
verified ·
1 Parent(s): 5c6f84b

Update download.py

Browse files
Files changed (1) hide show
  1. download.py +1 -1
download.py CHANGED
@@ -1,7 +1,7 @@
1
  import os
2
  from diffusers import DiffusionPipeline, FluxPipeline, AutoencoderTiny, AutoencoderKL
3
  from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast
4
-
5
  # Define models and their configurations (same as in app.py)
6
  models = {
7
  "FLUX.1-schnell": {
 
1
  import os
2
  from diffusers import DiffusionPipeline, FluxPipeline, AutoencoderTiny, AutoencoderKL
3
  from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast
4
+ import torch
5
  # Define models and their configurations (same as in app.py)
6
  models = {
7
  "FLUX.1-schnell": {