Hugging Face's logo

Spaces:
Djrango
/
qwen2vl-flux-mini-demo
Runtime error

App Files Community
4
erwold commited on
Commit
2645f74
·
1 Parent(s): d800f84

Initial Commit

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,8 +36,8 @@ class FluxInterface:
36
  tokenizer_two = T5TokenizerFast.from_pretrained(self.MODEL_ID, subfolder="flux/tokenizer_2")
37
 
38
  # Load VAE and transformer from flux folder
39
- vae = AutoencoderKL.from_pretrained(self.MODEL_ID, subfolder="flux")
40
- transformer = FluxTransformer2DModel.from_pretrained(self.MODEL_ID, subfolder="flux")
41
  scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(self.MODEL_ID, subfolder="flux/scheduler", shift=1)
42
 
43
  # Load Qwen2VL components from qwen2-vl folder
 
36
  tokenizer_two = T5TokenizerFast.from_pretrained(self.MODEL_ID, subfolder="flux/tokenizer_2")
37
 
38
  # Load VAE and transformer from flux folder
39
+ vae = AutoencoderKL.from_pretrained(self.MODEL_ID, subfolder="flux/vae")
40
+ transformer = FluxTransformer2DModel.from_pretrained(self.MODEL_ID, subfolder="flux/transformer")
41
  scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(self.MODEL_ID, subfolder="flux/scheduler", shift=1)
42
 
43
  # Load Qwen2VL components from qwen2-vl folder