Staticaliza commited on
Commit
927051d
·
verified ·
1 Parent(s): 2981d40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -7
app.py CHANGED
@@ -62,15 +62,10 @@ repo_customs = {
62
  "Realistic": None, #StableDiffusionXLPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
63
  "Anime": None, #StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
64
  "Pixel": None, #repo_pixel,
65
- "Large": StableDiffusion3Pipeline.from_pretrained(repo_large_path, transformer=repo_large_transformer, torch_dtype=torch.float16, use_safetensors=True,
66
- text_encoder_3=None,
67
- tokenizer_3=None),
68
  }
69
 
70
- repo_customs["Large"].scheduler = FlashFlowMatchEulerDiscreteScheduler.from_pretrained(
71
- repo_large_path,
72
- subfolder="scheduler",
73
- )
74
 
75
  # Functions
76
  def save_image(img, seed):
 
62
  "Realistic": None, #StableDiffusionXLPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
63
  "Anime": None, #StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
64
  "Pixel": None, #repo_pixel,
65
+ "Large": StableDiffusion3Pipeline.from_pretrained(repo_large_path, transformer=repo_large_transformer, torch_dtype=torch.float16, use_safetensors=True),
 
 
66
  }
67
 
68
+ repo_customs["Large"].scheduler = FlashFlowMatchEulerDiscreteScheduler.from_pretrained(repo_large_path, subfolder="scheduler")
 
 
 
69
 
70
  # Functions
71
  def save_image(img, seed):