Upload 2 files

app.py

@@ -131,7 +131,7 @@ def load_pipeline(repo_id: str, cn_on: bool, model_type: str, task: str, dtype_s
             pipe_i2i = pipeline_i2i.from_pipe(pipe, transformer=transformer, text_encoder_2=text_encoder_2, torch_dtype=dtype)
         elif ".safetensors" in repo_id or ".gguf" in repo_id: # from single file
             file_url = repo_id.replace("/resolve/main/", "/blob/main/").replace("?download=true", "")
-            if ".gguf" in file_url: transformer_model.from_single_file(file_url, subfolder="transformer",
+            if ".gguf" in file_url: transformer = transformer_model.from_single_file(file_url, subfolder="transformer",
                 quantization_config=GGUFQuantizationConfig(compute_dtype=dtype), torch_dtype=dtype, config=single_file_base_model)
             else: transformer = transformer_model.from_single_file(file_url, subfolder="transformer", torch_dtype=dtype, config=single_file_base_model)
             pipe = pipeline.from_pretrained(single_file_base_model, transformer=transformer, torch_dtype=dtype, token=hf_token, **kwargs)
@@ -160,7 +160,7 @@ def change_base_model(repo_id: str, cn_on: bool, disable_model_cache: bool, mode
     global pipe, pipe_i2i, taef1, good_vae, controlnet_union, controlnet, last_model, last_cn_on, last_task, last_dtype_str, dtype
     try:
         if not disable_model_cache and (repo_id == last_model and cn_on is last_cn_on and task == last_task and dtype_str == last_dtype_str)\
-            or ((not is_repo_name(repo_id) or not is_repo_exists(repo_id)) and not ".safetensors" in repo_id): return gr.update()
+            or ((not is_repo_name(repo_id) or not is_repo_exists(repo_id)) and not ".safetensors" in repo_id and not ".gguf" in repo_id): return gr.update()
         unload_lora()
         pipe.to("cpu")
         pipe_i2i.to("cpu")